POJ1015 Jury Compromise （动态规划 难）

http://poj.org/problem?id=1015

题意：给出n个人，每个人有两个值di和pi要求你从选出m个人是的总的D和P的差距最小，如果差距相同则输出D+P较大的，同时输出选择的m个人，按照升序排列。

做法：刚刚开始想了一个暴力的做法，就是dp[i][j][d][p]就是前i个人选j个人两种值的总和分别为d和p的时候的是否可行，吧第一维滚动掉。然后按着背包写了写，嗯发现找了一些样列来写发现好像是对的，一交果不其然直接WA，然后一瞬间发现答案可能是对的，但不过记录路径就有问题了。

其实正解应该是dp[i][j][k]表示前i个人选择j个人出来，差值为k，总和为dp[i][j][k]这样表示，在定义一个数组表示p[i][j][k]表示路径就可以了。然后就这样按着dp的思维写了半个小时，然后该了很久才弄出来。。。具体还是看代码吧，代码看起来还是挺清晰的。其中第一维还是可以滚动优化的。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<vector>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
int a[210],b[210],n,m;
int dp[210][22][810];
int p[210][22][810];
int main()
{
    int kase=0;
    while(~scanf("%d%d",&n,&m)&&n+m)
    {
        for(int i=1;i<=n;i++) scanf("%d%d",&a[i],&b[i]);
        memset(dp,-1,sizeof(dp));
        memset(p,0,sizeof(p));
        vector<int>v;
        for(int i=0;i<=n;i++) dp[i][0][400]=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m&&j<=i;j++)
            {
                for(int k=0;k<=800;k++)
                {
                    dp[i][j][k]=dp[i-1][j][k];
                    p[i][j][k]=p[i-1][j][k];
                    if(k>=a[i]-b[i]&&k-(a[i]-b[i])<=800&&dp[i-1][j-1][k-a[i]+b[i]]>=0)
                    {
                        if(dp[i-1][j-1][k-(a[i]-b[i])]+a[i]+b[i]>dp[i][j][k])
                        {
                            dp[i][j][k]=dp[i-1][j-1][k-(a[i]-b[i])]+a[i]+b[i];
                            p[i][j][k]=i;
                        }
                    }
                }
            }
        }
        int ans=1010,cnt=0,mr;
        for(int i=0;i<=800;i++)
        {
            if(dp[n][m][i]>=0){
                int t=abs(400-i);
                if(ans>t){
                    ans=t;
                    cnt=dp[n][m][i];
                    mr=i;
                }
                if(ans==t&&dp[n][m][i]>cnt){
                    mr=i;
                }
            }
        }
        int t1=n,t2=m,t3=mr,l=0,r=0;
        while(t2>0){
            int po=p[t1][t2][t3];
            l+=a[po],r+=b[po];
            v.push_back(po);
            t2--;t3=t3-a[po]+b[po];
            t1=p[po-1][t2][t3];
        }
        sort(v.begin(),v.end());
        printf("Jury #%d\n",++kase);
        printf("Best jury has value %d for prosecution and value %d for defence:\n",l,r);
        for(int i=0;i<v.size();i++) printf(" %d",v[i]);
        printf("\n\n");
    }
    return 0;
}