結果

問題 No.2596 Christmas Eve (Heuristic ver.)
コンテスト
ユーザー anonymously
提出日時 2023-12-25 07:07:04
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 16 ms / 1,224 ms
コード長 3,169 bytes
コンパイル時間 1,924 ms
コンパイル使用メモリ 186,720 KB
実行使用メモリ 6,548 KB
スコア 3,032,530
最終ジャッジ日時 2023-12-25 07:07:19
合計ジャッジ時間 14,072 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 125
権限があれば一括ダウンロードができます

ソースコード

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)];
	}
	
	//中央と幹の幅の差が小さい順に処理
	priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> Q;
	j=0;
	while(j<k){
	    Q.push(make_pair(woods[j][1].width-woods[j][3].width,j));
	    j++;
	}
	
	//先端を決める
	set<int> S; //確定した先端の番号
	while(!Q.empty()){
	    pair<int,int> q=Q.top();
	    Q.pop();
	    vector<wood> w=woods[q.second];
	    
		//中央２個と幹の高さ
		int height=0;
		for(int _=1;_<4;_++)height+=w[_].height;

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

		//暫定
		w[0]=wooden_head[c];
		
		while(wooden_head[++c].width<w[1].width){//先端の幅＜中央の幅の間ループ
			if(!S.count(wooden_head[c].id) && abs(height+wooden_head[c].height-avg)<abs(height+w[0].height-avg)){//高さの合計が平均に近ければ
				w[0]=wooden_head[c];
			}
			if(c+1>=n)break;
		}
		
		//出力
		for(int _=0;_<4;_++){
			if(_)cout << ' ';
			cout << w[_].id;
		}
		cout << endl;
		
		S.insert(w[0].id);
	}

	return 0;
}
0