题目:http://acm.hdu.edu.cn/showproblem.php?pid=2639
题意:
和01背包题意一样,只不过给了K,求第K优解。
在01背包的基础上,dp[i][j]]加了一维,变成dp[i][j][k],表示前i个物品容量为j时的第K大解。
加一层循环即可,然后把可能的解(前一状态,和放入该物品的状态)都分别存到a[],b[]数组中。
注意并列解是不是同一解。
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 1001
using namespace std;
int dp[1500][35],w[105],v[105],a[35],b[35];
int main(){
int t;
cin>>t;
while(t--){
memset(dp,0,sizeof(dp));
int n,V,K;
cin>>n>>V>>K;
for(int i=1;i<=n;i++)
cin>>v[i];
for(int i=1;i<=n;i++)
cin>>w[i];
for(int i=1;i<=n;i++){
for(int j=V;j>=w[i];j--){
for(int k=1;k<=K;k++){
a[k]=dp[j-w[i]][k]+v[i];
b[k]=dp[j][k];
}
a[K+1]=b[K+1]=-1;
int aa=1,bb=1,c=1;
while(c<=K && (a[aa]!=-1 || b[bb]!=-1)){
if(a[aa]>b[bb])
dp[j][c]=a[aa],++aa;
else
dp[j][c]=b[bb],++bb;
if(dp[j][c]!=dp[j][c-1])
++c; //看清楚如果并列是看做一个解还是两个解,这里是一个
}
}
}
cout<<dp[V][K]<<endl;
}
return 0;
}