Bone Collector II HDU - 2639 01揹包求K大值

題意：

給出商品價值、體積，以及揹包能裝的最大容量，求出能裝入的第K大的價值。

如能裝入的價值爲20 12 2，那麼第2大的就是12.

思路：

給0-1揹包增加一個維度，dpjk代表裝入容量爲j時第k大的價值。

我們每次合併 dp[j][k]和dp[j-w[i]][k]+v[i]，取前k大進行更新。

代碼：

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
int dx[10]= {-1,1,0,0};
int dy[10]= {0,0,-1,1};
const int maxn=1000+10;
int v[maxn],w[maxn];
int dp[maxn][50];
int a[maxn],b[maxn];
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        memset(dp,0,sizeof(dp));
        int N,V,K;
        cin>>N>>V>>K;
        rep(i,1,N)
        {
           cin>>v[i];
        }
        rep(i,1,N)
        {
           cin>>w[i];
        }
        rep(i,1,N)
        {
            dep(j,V,w[i])
            {
                rep(k,1,K)
                {
                    a[k]=dp[j][k];
                    b[k]=dp[j-w[i]][k]+v[i];
                }
                a[K+1]=b[K+1]=-1;
                int l=1,r=1,cnt=1;
                while((a[l]>=0||b[l]>=0)&&cnt<=K)
                {
                    if(a[l]>b[r])
                    {
                        dp[j][cnt]=a[l++];
                    }
                    else
                    {
                        dp[j][cnt]=b[r++];
                    }
                    if(dp[j][cnt]!=dp[j][cnt-1])cnt++;
                }
            }
        }
        cout<<dp[V][K]<<endl;
    }
}

參考：https://blog.csdn.net/Tc_To_Top/article/details/49872403?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task