acwing280.陪审团 01背包

link
01背包

思路

f[i][j][k]表示在前i个人中选出j个人，此时辩方总分与控方总分差为k时，二者和的最大值
转移方程即为：f[i][j][k] = max(f[i][j][k],f[i-1][j-1][k-a[i]+b[i]]+a[i]+b[i])

1. 维护 $f$ 数组
   由于是绝对值，所以要用一个偏移量delta，由于评分都在0-20之间，且最多选20人，也就是说abs(k)最多为400，于是这里取delta = 400，所以 k 的范围就是 $0\le k\le 800$。
2. 找出最小差值k
   显然就是最接近400的数。
3. 找出对应的方案入选的人
   方法是反过来推。首先从f[i][j][v](i=n,j=m,这里的v是上面找出来的最小差值加上base)开始，判断他是由哪一种情况得来（选与不选），若是不选，直接将i–；若选，记录人选，将v减掉该人的差值，
   再i–，j–。
   也正是因为要找出对应方案入选的人，本题不能优化为二维数组。

代码

int f[210][21][805];//f[i][j][k]表示在前i个人中选出j个人，此时辩方总分与控方总分差为k时，二者和的最大值
int a[210], b[210];
void solve() {
    int n, m;
    int t = 0;
    while(cin >> n >> m && n) {
    	for(int i = 1; i <= n; i++) {
    		cin >> a[i] >> b[i];
    		// a[i] += 20;
    	}
    	cout << "Jury #" << ++t << endl;
    	memset(f, -0x3f,sizeof(f));
    	f[0][0][400] = 0;
    	for(int i = 1; i <= n; i++) {
    		for(int j = 0; j <= m; j++) {
    			for(int k = 0; k <= 800; k++) {
    				int p = k - (a[i] - b[i]);
    				f[i][j][k] = f[i-1][j][k];
    				if(j - 1 < 0) continue;
    				if(p < 0 || p > 800) continue;
    				f[i][j][k] = max(f[i][j][k], f[i-1][j-1][p] + a[i] + b[i]);
    			}
    		}
    	}
    	int v = 0;
    	while(f[n][m][400-v] < 0 && f[n][m][400 + v] < 0) v++;
    	if(f[n][m][400-v] > f[n][m][400+v]) v = 400 - v;
    	else v = 400 + v;
    	// cout << f[n][m][v] << endl;
    	int ans1, ans2;
    	ans1 = (f[n][m][v] + abs(v - 400)) / 2;
    	ans2 = f[n][m][v] - ans1;
    	if(v < 400) 
    		swap(ans1, ans2);
    	stack<int> s;
    	int j = m;
    	for(int i = n; i >= 1; i--) {
    		if(f[i][j][v] == f[i-1][j][v]) {
    			continue;
    		}
    		s.push(i);
    		v -= a[i] - b[i];
    		j--;
    	}
    	printf("Best jury has value %d for prosecution and value %d for defence:\n",ans1, ans2);
    	cout << ' ';
    	while(!s.empty()) {
    		cout << s.top() << ' ';
    		s.pop();
    	}
    	cout << endl << endl;
    }
}