hdu Bone Collector II(01背包第K优解)

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值