题意:传送门
题解:在dp数组上多加上一维记录第k优解,之后利用类似归并排序那样进行递推下去即可。
附上代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e2+5;
const int maxv=1e3+5;
const int maxk=30+5;
int T,n,v,k;
int val[maxn],vol[maxn];
int dp[maxv][maxk];
int a[maxk],b[maxk];
int main()
{
scanf("%d",&T);
while(T--){
scanf("%d%d%d",&n,&v,&k);
for(int i=1;i<=n;i++){
scanf("%d",&val[i]);
}
for(int i=1;i<=n;i++){
scanf("%d",&vol[i]);
}
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++){
for(int j=v;j>=vol[i];j--){
for(int l=1;l<=k;l++){
a[l]=dp[j-vol[i]][l]+val[i];
b[l]=dp[j][l];
}
a[k+1]=-1;b[k+1]=-1;
int ida=1,idb=1,idc=1;
while(idc<=k&&(a[ida]!=-1||b[idb]!=-1)){
if(a[ida]>b[idb]){
dp[j][idc]=a[ida++];
}else{
dp[j][idc]=b[idb++];
}
if(dp[j][idc]!=dp[j][idc-1]){
idc++;
}
}
}
}
printf("%d\n",dp[v][k]);
}
return 0;
}