Jury Compromise POJ - 1015 （二维dp）

在遥远的国家佛罗布尼亚，嫌犯是否有罪，须由陪审团决定。陪审团是由法官从公众中挑选的。先随机挑选n个人作为陪审团的候选人，然后再从这n个人中选m人组成陪审团。选m人的办法是： 控方和辩方会根据对候选人的喜欢程度，给所有候选人打分，分值从0到20。为了公平起见，法官选出陪审团的原则是：选出的m个人，必须满足辩方总分和控方总分的差的绝对值最小。如果有多种选择方案的辩方总分和控方总分的之差的绝对值相同，那么选辩控双方总分之和最大的方案即可。

Input

输入包含多组数据。每组数据的第一行是两个整数n和m，n是候选人数目，m是陪审团人数。注意，1<=n<=200, 1<=m<=20 而且 m<=n。接下来的n行，每行表示一个候选人的信息，它包含2个整数，先后是控方和辩方对该候选人的打分。候选人按出现的先后从1开始编号。两组有效数据之间以空行分隔。最后一组数据n=m=0

Output

对每组数据，先输出一行，表示答案所属的组号,如 'Jury #1', 'Jury #2', 等。接下来的一行要象例子那样输出陪审团的控方总分和辩方总分。再下来一行要以升序输出陪审团里每个成员的编号，两个成员编号之间用空格分隔。每组输出数据须以一个空行结束。

Sample Input

4 2 
1 2 
2 3 
4 1 
6 2 
0 0 

Sample Output

Jury #1 
Best jury has value 6 for prosecution and value 4 for defence: 
 2 3 

Hint

If your solution is based on an inefficient algorithm, it may not execute in the allotted time.

#include <iostream>
#include <string>
#include <set>
using namespace std;
int dp[30][810], path[30][810];
int p[210], d[210];
int a[210], b[210];
int n, m;
set<int> res;

bool isfind(int tmpi, int tmpj, int x)
{
	if (tmpi == 0) return false;
	return path[tmpi][tmpj] == x ? true : isfind(tmpi - 1, tmpj - a[path[tmpi][tmpj]], x);
}

void print(int i, int j)
{
	int ans1, ans2;

	ans1 = ans2 = 0;
	while (i)
	{
		int temp = path[i][j];
		ans1 += p[temp], ans2 += d[temp];
		res.insert(temp);
		j -= a[temp];
		i--;
	}
	printf("Best jury has value %d for prosecution and value %d for defence:\n", ans1, ans2);
	set<int>::iterator it = res.begin();
	for (; it != res.end(); it++)
		printf(" %d", *it);
	printf("\n\n");
	return;
}

int main()
{
	int cases = 0;

	while (cin >> n >> m && n && m)
	{
		for (int i = 1; i <= n; i++)
		{
			cin >> p[i] >> d[i];
			a[i] = p[i] - d[i];
			b[i] = p[i] + d[i];
		}
		memset(dp, -1, sizeof(dp));
		dp[0][20 * m] = 0;
		memset(path, 0, sizeof(path));
		
		int num = 2 * 20 * m;
		for(int i = 1; i <= m; i++)
			for (int j = 0; j <= num; j++)
				for(int k = 1; k <= n; k++)
					if (j - a[k] >= 0 && j - a[k] <= num && dp[i - 1][j - a[k]] != -1 && isfind(i - 1, j - a[k], k) == false)
						if(dp[i][j] < dp[i - 1][j - a[k]] + b[k])
						{
							dp[i][j] = dp[i - 1][j - a[k]] + b[k];
							path[i][j] = k;
						}
		printf("Jury #%d\n", ++cases);
		res.clear();
		for (int j = 0; j <= m * 20; j++)
		{
			if (dp[m][m * 20 + j] > dp[m][m * 20 - j])
			{
				print(m, m * 20 + j);
				break;
			}
			else if (dp[m][m * 20 - j] != -1)
			{
				print(m, m * 20 - j);
				break;
			}
		}
	}

}