結果

問題 No.2596 Christmas Eve (Heuristic ver.)
コンテスト
ユーザー anonymously
提出日時 2023-12-24 23:23:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,765 bytes
コンパイル時間 1,807 ms
コンパイル使用メモリ 176,740 KB
実行使用メモリ 6,676 KB
スコア 0
最終ジャッジ日時 2023-12-24 23:23:36
合計ジャッジ時間 26,518 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 17 RE * 108
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
//#define DEBUG

typedef struct wood{
	int id;
	int width;
	int height;
}wood;

bool compare(wood a, wood b){
	if(a.width!=b.width){
		return a.width<b.width;
	}else{
		return a.id<b.id;
	}
}

/*
 * とりあえずACを取得する為の解答
 */
int main(){
	int n,k;
	cin >> n >> k;
	
	//高さの合計
	int sum=0;
	
	//「木の先端」
	vector<wood> wooden_head(n);
	
	//「木の先端」の幅
	for(int i=0;i<n;i++){
		wooden_head[i].id=i+1;
		cin >> wooden_head[i].width;
	}
	
	//「木の先端」の高さ
	for(int i=0;i<n;i++){
		cin >> wooden_head[i].height;
		sum+=wooden_head[i].height;
	}
	
	//ソート
	sort(wooden_head.begin(),wooden_head.end(),compare);
	
	//「木の中央」
	vector<wood> wooden_central(2*n);
	
	//「木の中央」の幅
	for(int i=0;i<2*n;i++){
		wooden_central[i].id=i+1;
		cin >> wooden_central[i].width;
	}
	
	//「木の中央」の高さ
	for(int i=0;i<2*n;i++){
		cin >> wooden_central[i].height;
		sum+=wooden_central[i].height;
	}
	
	//ソート
	sort(wooden_central.begin(),wooden_central.end(),compare);
	
	//「木の幹」
	vector<wood> wooden_trunk(n);
	
	//「木の幹」の幅
	for(int i=0;i<n;i++){
		wooden_trunk[i].id=i+1;
		cin >> wooden_trunk[i].width;
	}
	
	//「木の幹」の高さ
	for(int i=0;i<n;i++){
		cin >> wooden_trunk[i].height;
		sum+=wooden_trunk[i].height;
	}

	//ソート
	sort(wooden_trunk.begin(),wooden_trunk.end(),compare);
	
	//木の高さの平均
	int avg=(sum+n/2)/n;
	
#ifdef DEBUG
	for(wood w:wooden_head){
		cout << w.id << ' ';
	}
	cout << endl;
	for(wood w:wooden_central){
		cout << w.id << ' ';
	}
	cout << endl;
	for(wood w:wooden_trunk){
		cout << w.id << ' ';
	}
	cout << endl;
#endif
	
	vector<vector<wood>> woods(k,vector<wood>(4));
	//幹は幅小⇒大、中央は幅大⇒小に確定させていく
	int j=0;
	while(j<k){
		woods[j][3]=wooden_trunk[j];
		j++;
		woods[k-j][2]=wooden_central[2*(n-j)+1];
		woods[k-j][1]=wooden_central[2*(n-j)];
	}
	
	//先端を決める
	j=0;
	while(j<k){
		//中央２個と幹の高さ
		int height=0;
		for(int _=1;_<4;_++)height+=woods[j][_].height;

		//先端探索開始位置
		int c=(j?woods[j-1][0].id:0);
		while(wooden_head[c].width<=woods[j][3].width){//先端の幅＞幹の幅になるまでループ
			c++;
		}

		//暫定
		woods[j][0]=wooden_head[c];
		
		while(wooden_head[++c].width<woods[j][1].width){//先端の幅＜中央の幅の間ループ
			if(abs(height+wooden_head[c].height-avg)<abs(height+woods[j][0].height-avg)){//高さの合計が平均に近ければ
				woods[j][0]=wooden_head[c];
			}
		}
		j++;
	}

	j=0;
	while(j<k){
		for(int _=0;_<4;_++){
			if(_)cout << ' ';
			cout << woods[j][_].id;
		}
		cout << endl;
		j++;
	}
	
	return 0;
}
0