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

HDU2639：Bone Collector II(01背包第k优解)

http://acm.hdu.edu.cn/showproblem.php?pid=2639

这题是上面那道的续题，求的是第 K 优解。

在01的基础上多加一维，dp[v][k]，表示在v空间下第k大的价值。

遍历 第 i 个物品在体积为 V 大小的背包，有拿与不拿两种情况。每种情况又与上一步 K 个解对应。一共是2 * K种解。取出其中前 K 大的解。即为这一步中的前 K 大的解。

在取 K 大数的时候，我用 set 去掉了重复的数，然后放进优先队列排序，取出前 K 个数，不足的话即为 0。

下面是 AC 代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<set>
#include<queue>
using namespace std;
int w[1005];
int v[1005];
int dp[1005][35];
int main()
{
    int t,n,m,k;
    set<int> s;
    priority_queue <int> q;
    scanf("%d",&t);
    while(t--)
    {
        memset(w,0,sizeof(w));
        memset(v,0,sizeof(v));
        memset(dp,0,sizeof(dp));
        scanf("%d%d%d", &n, &m, &k);
        for(int i=1; i<=n; i++)
            scanf("%d",&v[i]);
        for(int i=1; i<=n; i++)
            scanf("%d",&w[i]);
        for(int i=1; i<=n; i++)
        {
            for(int j=m; j>= w[i]; j--)
            {
            	s.clear();
            	for (int g = 1; g <= k; g++) { //遍历 前 k 大的数 
            		s.insert(dp[j - w[i]][g] + v[i]);
                    s.insert(dp[j][g]);
				}
				for (set<int>::iterator iter = s.begin(); iter != s.end(); iter++) { //用set去掉重复的数 
					q.push(*iter);
				}
                for (int g = 1; g <= k && !q.empty(); g++) {//用优先队列取前K大的数 
                	dp[j][g] = q.top();
                	q.pop();
				}
				while (!q.empty()) q.pop();//清空队列 
            }
        }
        printf("%d\n",dp[m][k]);
    }
    return 0;
}