dp[i][j]=k 表示選取 i 個人,其評價差之和爲 j 的所有方案中總評價之和最大爲 k
三重循環,找出各種人數在各個評價差上的最大總評價之和,評價差最多就m202,因爲是-20m到20m之間,所以區間平移,計算出來最大區間也不過800。
錯誤做法中,path存儲路徑有缺陷,當要選擇6個人時,碰到選擇5個人的方案有兩種評價差的和一樣小,總評價的和一樣大,這是path只會記錄一種方案,而如果答案恰巧是另一種方案加上記錄的方案中的其中一個人,那麼就不會選到, 所以開三維數組,每次都記錄全部的路徑,而且把選人的循環放最外層,就不用判斷重複選人了
#include <algorithm>
#include <iostream>
#include <stdio.h>
#include <string>
#include <cstring>
#include <vector>
#include <set>
#include <math.h>
#define eps 1e-14
#define pi acos(-1)
#define ll long long
#define RD T*(rand()*2-RAND_MAX)
#define Drand (long double)rand()/RAND_MAX
#define LINF 0x7f7f7f7f7f7f7f7f
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=1005;
const long long mod=1e18;
ll MOD(ll a,ll m){return a>m?a%m+m:a;}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
int dp[50][1000];
vector<int>path[21][801];
int cha[1000],he[1000];
int main()
{
// freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
int n,m;
int cas=1;
while(~scanf("%d%d",&n,&m) && n && m){
memset(dp,-1,sizeof dp);
memset(path,0,sizeof path);
for (int i=1; i<=n; i++) {
int t1,t2;
scanf("%d%d",&t1,&t2);
cha[i] = t1-t2;
he[i] = t1+t2;
}
int offset=m*20;
dp[0][offset]=0;
for(int i=1;i<=n;i++){
for(int j=m-1;j>=0;j--){
for(int k=0;k<=2*offset;k++){
if(dp[j][k]>=0){
if(dp[j+1][k+cha[i]]<=dp[j][k]+he[i]){
dp[j+1][k+cha[i]]=dp[j][k]+he[i];
path[j+1][k+cha[i]]=path[j][k];
path[j+1][k+cha[i]].push_back(i);
}
}
}
}
}
int pos=0;
for(pos=0;pos<=offset;pos++){
if(dp[m][offset-pos]>=0 || dp[m][offset+pos]>=0)break;
}
pos=dp[m][offset-pos] > dp[m][offset+pos] ? offset-pos : offset+pos;
printf("Jury #%d\n",cas++);
printf("Best jury has value %d for prosecution and value %d for defence:\n",(dp[m][pos]+pos-offset)/2,(dp[m][pos]-pos+offset)/2);
for(int i=0;i<path[m][pos].size();i++){
printf(" %d",path[m][pos][i]);
}
printf("\n\n");
}
return 0;
}