SRM diary(Sigmar)

TCO2012 Round2B 550 HeavyBooks

TopCoder Open | 17:55 |

Problem Statement

コーディングフェーズ

うーん

一番重いのを押し付けあうとしか思えない

それより軽いのを渡しても得する場面が全く思いつかない

一番重いのを押し付けあうとすると、最初にTomekがどの本を入れるか決めるだけの問題

重い順に見ていけばどちらが持つかは一意に決まるので、重い順に本を採用するかしないかでDPすればOKかな

書こうとする

添字とDPの更新条件式がカオスになる

デバッグで最終的に40分もかかった

提出

ソースコード

DP更新条件式のカオスなところは直しました

今回はpairを使えば比較的ラクに更新ができる気がする

本番でもpairを使っていたのになぜカオスになったんだ・・・

const int INF=1000000000;

class HeavyBooks {
public:
	vector <int> findWeight(vector <int> books, vector <int> moves) {
		vector <int> res;
		int n=books.size();
		int m=moves.size();

		vector <int> b(moves[0], 1);
		for(int i=1; i<m; i++) {
			int idx=0;
			for(int j=0; j<moves[0]; j++) {
				if(b[j]==(i&1)) {
					b[j]=1-b[j];
					idx++;
					if(idx==moves[i]) break;
				}
			}
		}

		pair <int, int> difsum[52][52];
		for(int i=0; i<52; i++) for(int j=0; j<52; j++) {
			difsum[i][j]=make_pair(-INF, 0);
		}
		difsum[0][0]=make_pair(0, 0);
		sort(books.begin(), books.end(), greater <int>());
		for(int d=0; d<n; d++) {
			for(int bi=0; bi<moves[0]; bi++) {
				pair <int, int> cand;
				cand=difsum[d][bi];
				difsum[d+1][bi]=max(difsum[d+1][bi], cand);
				if(b[bi]) {
					cand=difsum[d][bi];
					cand.first+=books[d]; cand.second+=books[d];
					difsum[d+1][bi+1]=max(difsum[d+1][bi+1], cand);
				} else {
					cand=difsum[d][bi];
					cand.first-=books[d]; cand.second+=books[d];
					difsum[d+1][bi+1]=max(difsum[d+1][bi+1], cand);
				}
			}
		}
		pair <int, int> best(-INF, 0);
		for(int i=0; i<=n; i++) {
			best=max(best, difsum[i][moves[0]]);
		}
		
		res.push_back((best.second-best.first)/2);
		res.push_back((best.first+best.second)/2);
		return res;
	}
};