Bone Collector II HDU - 2639 (01背包+k优解)

题意:传送门

题解:在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;
}

猜你喜欢

转载自blog.csdn.net/zhouzi2018/article/details/86555311