POJ1015 Jury Compromise

参考博客

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <queue>
#include <vector>
#include <map>
using namespace std;
typedef long long ll;
typedef double db;
const int mod=1e9+7;
const int maxn=1e6;
const double eps=0.00000001;
int n,m,dp[25][850],d[205][25][1111],a[205],b[205],suma,sumb;
vector<int>ssc;
//dp[i][j]代表已选有i个候选人，差值为j的和是多少。
//d[i][j][k]，i代表包含第i个人以及前面的人，j代表已选有j个人，k差值为k，d代表当前状态下选择的人选的编号。 
void get_path(int i,int j,int k){//递归求路径 
	if(j==0) return;
	int last=d[i][j][k];
	get_path(last-1,j-1,k-(a[last]-b[last]));
	ssc.push_back(last);
	suma+=a[last],sumb+=b[last];
}
int main(){
	int T=0;
	while(scanf("%d%d",&n,&m)==2&&n&&m){
		suma=sumb=0;
		for(int i=1;i<=n;++i) scanf("%d%d",&a[i],&b[i]);
		memset(dp,0xcf,sizeof(dp));//初始化无穷小 
		//int fix=m*20;
		dp[0][400]=0;//防止出现下表为负数，将数组进行平移，原本dp[0][0]更为dp[0][400];
		for(int i=1;i<=n;++i){
			for(int j=0;j<=m;++j)
				for(int k=0;k<=800;++k) d[i][j][k]=d[i-1][j][k];//更新 
			for(int j=m;j>0;--j)
				for(int k=0;k<=800;++k){
					if(k-(a[i]-b[i])<0||k-(a[i]-b[i])>800) continue;//超出范围 
					if(dp[j][k]<dp[j-1][k-(a[i]-b[i])]+a[i]+b[i]){
						dp[j][k]=dp[j-1][k-(a[i]-b[i])]+a[i]+b[i];
						d[i][j][k]=i;//更新路径 
					}
				}
		}
		int ans=0;
		for(int i=0;i<=400;++i){
			if(dp[m][400+i]>=0||dp[m][400-i]>=0){
				ans=dp[m][400+i]>=dp[m][400-i]?400+i:400-i;//找差值最小且和最大 
				break;
			}
		}
		ssc.clear();
		get_path(n,m,ans);
		printf("Jury #%d\n", ++T);
		printf("Best jury has value %d for prosecution and value %d for defence:\n", suma, sumb);
		for (int i=0;i<ssc.size();i++) printf(" %d",ssc[i]);
		printf("\n\n");
	}
	return 0;
}