POJ 1015 Jury Compromise (动态规划)

dp[i][j]代表选了i个人，D(J)-P(J)的值为j的状态下，D(J)+P(J)的最大和。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>

using namespace std;

const int MAXN = ;

int n, m;
int dp[][MAXN];
int path[][MAXN];
int P[], D[];
int sub[], sum[];
int maxval, fix;
int ans[];

bool IfHave( int i, int j, int k )
{
    while ( i >  )
    {
        if ( path[i][j] == k ) return true;
        j -= sub[ path[i][j] ];
        --i;
    }
    return false;
}

void DP()
{
    memset( dp, -, sizeof(dp) );
    dp[][+fix] = ;
    for ( int i = ; i <= m; ++i )
        for ( int j = -fix; j <= fix; ++j )
        {
            if ( dp[i - ][j + fix] >=  )
            {
                for ( int k = ; k <= n; ++k )
                {
                    int newdp = dp[i - ][ j+fix ]+sum[k];
                    int &curdp = dp[i][ j+fix+sub[k] ];
                    if ( !IfHave( i-, j+fix, k) && newdp > curdp )
                    {
                        path[i][j+fix+sub[k]] = k;
                        curdp = newdp;
                    }
                }
            }
        }

    return;
}

void getPath( int i, int j )
{
    int cnt = ;
    while ( i >  )
    {
        ans[cnt++] = path[i][j];
        j -= sub[ path[i][j] ];
        --i;
    }
    return;
}

int main()
{
    int cas = ;
    while ( scanf( "%d%d", &n, &m ) ==  && (n || m) )
    {
        for ( int i = ; i <= n; ++i )
        {
            scanf( "%d%d", &P[i], &D[i] );
            sub[i] = P[i] - D[i];
            sum[i] = P[i] + D[i];
        }

        fix =  * m;
        maxval =  * fix;

        DP();

        int anssub, anssum;
        for ( int i = ; i <= fix; ++i )
        if ( dp[m][fix-i] >=  || dp[m][fix+i] >=  )
        {
            if ( dp[m][fix-i] > dp[m][fix+i] )
            {
                anssub = fix-i;
                anssum = dp[m][fix-i];
            }
            else
            {
                anssub = fix+i;
                anssum = dp[m][fix+i];
            }
            break;
        }

        getPath( m, anssub );
        int sum1 = , sum2 = ;
        for ( int i = ; i < m; ++i )
        {
            sum1 += P[ ans[i] ];
            sum2 += D[ ans[i] ];
        }
        printf( "Jury #%d\n", ++cas );
        printf( "Best jury has value %d for prosecution and value %d for defence:\n", sum1, sum2 );
        sort( ans, ans + m );
        for ( int i = ; i < m; ++i )
            printf( " %d", ans[i] );
        puts("\n");
    }
    return ;
}