陪审团【01背包】【输出路径】

题意

n个人选m个人，每个人有p[i],d[i]的值，要选出这m个人的所有(p值和)-(d值和)最小，并且p值和+d值和最大。并且输出方案。

思路

组合最优化问题，背包问题。

p-d最小，把这个约束定为一维度，

选择m个数，把这一约束定为一维度。

f[i] [j] [k] : 前i个数，选了j个数,p与d的差值为k,的和值最大。

因为我们要输出路径，不能滚动数组优化，另外要把负值外加偏移量。

要保证j的维度决定值最小，要判断这个值的正负。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define mst(s,_s) memset(s, _s, sizeof(s))
const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF = 0x3f3f3f3f;
const int N = 1e6+100,mov=400;
int T,n,m;
int p[210],d[210];
int f[210][30][810];
int path[N],cnt;
int count1=0;
int main() {
int T=1;
 while (scanf("%d%d", &n, &m), n || m)
{
        for (int i = 1; i <= n; i ++ ) scanf("%d%d", &p[i], &d[i]);

        memset(f,-0x3f,sizeof f);
        f[0][0][mov]=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<=m;j++)
            {
                for(int k=0;k<810;k++)
                {
                    f[i][j][k]=f[i-1][j][k];
                    int t=k-(p[i]-d[i]);
                    if(t<0 || t>=810) continue;
                    if(j<1) continue;
                    f[i][j][k]=max(f[i][j][k],f[i-1][j-1][t] + p[i]+d[i]);
                    
                }
            }
        }
        
        int minv=0;
        while(f[n][m][mov+minv]<0 && f[n][m][mov-minv] <0) minv++;
        if(f[n][m][mov+minv] < f[n][m][mov-minv]) minv=mov-minv;
        else
            minv=minv + mov;
        cnt=0;
        //int i = n, j = m, k = minv;
        int j=m;
        for(int i=n;i;i--)
        {
            if(f[i][j][minv]==f[i-1][j-1][minv-(p[i]-d[i])] + p[i]+d[i]){

                path[++cnt]=i;
                j--;
                if(j==0) break;
                minv-=p[i]-d[i];
            }
                
        }
        // while (j)
        // {
        //     if (f[i][j][k] == f[i - 1][j][k]) i -- ;
        //     else
        //     {
        //         path[++cnt] = i;
        //         k -= (p[i] - d[i]);
        //         i --, j -- ;
        //     }
        // }
        int sp = 0, sd = 0;
        for (int i = 1; i <= cnt; i ++ )
        {
            sp += p[path[i]];
            sd += d[path[i]];
        }

        printf("Jury #%d\n", T ++ );
        printf("Best jury has value %d for prosecution and value %d for defence:\n", sp, sd);

        sort(path+1, path +1+ cnt);
        for (int i = 1; i <= cnt; i ++ ) printf(" %d", path[i]);
        puts("\n");
    }
    



    return 0;
}