poj1015 Jury Compromise(01背包+输出路径)

题目

给定n(1<=n<=200)个人的信息，

第i个人的信息由两个分数组成，控方对其评分pi，辩方对其评分di(0<=pi,di<=20)

你需要选出恰好m(1<=m<=min(n,20))个人，

使得辩方总分和控方总分的差的绝对值最小。

若存在多种方案总分之差绝对值相同，选辩控双方总分之和最大的。

输出最终的控方总分和辩方总分及选了哪几个人。

思路来源

https://www.cnblogs.com/FrankChen831X/p/10422233.html

题解

uva323是本题的数据加强版，然而通过了思路来源提供的样例应该就没问题

两个变量的，一般下标放一个，值放一个

一个变量是abs(x-y)，一个变量是x+y，前者不便维护值，所以加偏移量放下标

dp[j][k]表示前i个人 选j个人 控-辩的分差为k时 的最大控+辩的和

转移时，01背包，同时转移人数和分数差，更新最大和即可

 

考虑记录路径，如果仅开二维数组记录前驱时，可能会出错，原因是01背包刷新后记录相同前驱

所以，要么开三维pre[i][j][k]数组记录前驱，代表前i个人中选j个人控辩差为k的前驱状态，往回倒推

要么开二维path[j][k]的vector实时记录路径，即选j个人控辩差为k时的最优方案

由于本题数据约束，具体实现时，三维pre的空间大时间略小，vector反之，此处取后者

代码

#include<iostream>
#include<cstdio> 
#include<vector>
#include<cstring>
using namespace std;
const int N=22,M=805,off=400,INF=0x3f3f3f3f;
//dp[j][k]:前i个人 选j个人 控-辩的分差为k时 的最大控+辩的和
//例如:dp[i][j][k]和dp[i][j-1][k-sub]都选取了i 且pre[i][j][k]=pre[i-1][j-1][k-sub]时 
//表现在滚动数组中 dp[j][k]和dp[j-1][k-sub]都选择了i 则选取重复
//事实上 是因为 把第一维滚动掉时 无法区分dp[i][j-1][k-sub]和dp[i-1][j-1][k-sub] 从而无法记录前驱的路径  
//所以用二维vector实时记录路径 或者开三维pre来记录前驱 
vector<int>path[N][M];
int cas,n,m,a,b,sub,add,dp[N][M];
bool upd(int &x,int y)
{
	if(x<y){x=y;return 1;}
	return 0;
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		if(!n&&!m)break;
		memset(dp,-INF,sizeof dp);
		for(int i=1;i<=m;++i)
		for(int j=0;j<M;++j)
		path[i][j].clear();
		dp[0][off]=0;
		for(int i=1;i<=n;++i)
		{
			scanf("%d%d",&a,&b);
			sub=a-b;add=a+b;
			int up=min(i,m);
			for(int j=up;j>=1;--j)
			{
				for(int k=0;k<M;++k)
				{
					if(k-sub<0||k-sub>=M||dp[j-1][k-sub]<0)continue;
					if(upd(dp[j][k],dp[j-1][k-sub]+add))
					{
						path[j][k]=path[j-1][k-sub];
						path[j][k].push_back(i);
					} 
				}
			}
		}
		sub=add=-1;
		int y; 
		for(int k=0;k<=off;++k)
		{
			if(dp[m][off-k]>=0||dp[m][off+k]>=0)
			{
				if(upd(add,dp[m][off-k]))y=off-k;
				if(upd(add,dp[m][off+k]))y=off+k;
				sub=y-off;//注意差值的符号 
				break;
			} 
		}
		printf("Jury #%d\n",++cas);
		printf("Best jury has value %d for prosecution and value %d for defence:\n",(add+sub)/2,(add-sub)/2);
		for(int i=0;i<m;++i)
		printf(" %d",path[m][y][i]);
		printf("\n\n");
	}
	return 0;
}