問題
http://community.topcoder.com/stat?c=problem_statement&pm=13211&rd=15857
・様々な高さを持つ建物が与えられる。
・建物の高さを1変えるのにコストが1かかる。
・同じ建物がm個必ずあるようにしたい。
・建物の数をnとしたとき、mは1~nすべての値としてそれぞれの最小コストのXORを求める。
解き方
・mを1~nそれぞれについて調べることとし、各建物の長さについて答えを求める。
・求め方として2分探索?でも最小値を求めるので使えなさそう。
・全探索では4000*4000*4000かかりそう。
→他の人のコードをみてみる
・mを固定した時、どの長さにするかは各建物の元々の長さにすればよい、、ことを単純ながら見落としてしまった。
・あとは各建物の長さに固定した時のコスト計算をO(1)とするために、累積和をとっておけばよい。
・反省:最適解探索の検討がまだまだ足りない。
コード
using namespace std;
#define all(c) (c).begin(),(c).end()
#define FORE(i,d,e) for(int i=d;i<e;i++)
#define FOR(i,s,e) for (int i = int(s); i != int(e); i++)
#define FORIT(i,c) for (typeof((c).begin()) i = (c).begin(); i != (c).end(); i++)
#define ISEQ(c) (c).begin(), (c).end()
class BuildingHeights {
public:
int minimum(vector<int> heights) {
int n=heights.size();
sort(all(heights));
int sum[n+1];
sum[0]=0;
FORE(i,1,n+1)sum[i]=sum[i-1]+heights[i-1];
int ret=0;
FORE(i,1,n+1){
int cur=1e+9;
FORE(j,i-1,n)cur=min(cur,heights[j]*i-(sum[j+1]-sum[j+1-i]));
ret^=cur;
}
return ret;
}
};
#define all(c) (c).begin(),(c).end()
#define FORE(i,d,e) for(int i=d;i<e;i++)
#define FOR(i,s,e) for (int i = int(s); i != int(e); i++)
#define FORIT(i,c) for (typeof((c).begin()) i = (c).begin(); i != (c).end(); i++)
#define ISEQ(c) (c).begin(), (c).end()
class BuildingHeights {
public:
int minimum(vector<int> heights) {
int n=heights.size();
sort(all(heights));
int sum[n+1];
sum[0]=0;
FORE(i,1,n+1)sum[i]=sum[i-1]+heights[i-1];
int ret=0;
FORE(i,1,n+1){
int cur=1e+9;
FORE(j,i-1,n)cur=min(cur,heights[j]*i-(sum[j+1]-sum[j+1-i]));
ret^=cur;
}
return ret;
}
};