HDU2639 Bone Collector II 背包K优解

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

【思路】

背包K优解是指背包解的第K最值。

跟普通背包的思路差不多，不同的是多了个维护K个最值。

所以复杂度为O(nvk)，其中n为背包数目，v为背包容量，k为所求第K优解。

普通的O(nv)的背包，空间复杂度为O(nv)，优化后的空间复杂度为O(v)。

所以，K优解的空间复杂度为O(nvk)，优化后的空间复杂度为O(vk)，道理是一样的。

所以dp[v][k]，对于每一个v都保存着k个值，从大到小排列。

还是一样的二重循环，然后加多一重循环，对于dp[j]，需要计算dp[j-volume[i]]+value[i]。

然后对dp[j-volume[i]]+value[i]的k个值和dp[j]的k个值进行合并。合并后得到的新的k个最值保存回dp[j]中。

这样就可以实现背包K优解的计算。

最后输出dp[v][k]（如果保存时从dp[v][1]开始保存）。

相对来说思路还是很清晰的，代码实现也很简洁。

【代码】

#include <iostream>
using namespace std;

const int maxn = 100;
const int maxv = 1000;
const int maxk = 30;

int dp[maxv+5][maxk+5];
int a[maxk+5], b[maxk+5];
int value[maxn+5];
int volume[maxn+5];

int main()
{
	int t;
	int n, v, k;
	int i, j, l, x, y, z;
	scanf("%d", &t);
	while(t--)
	{
		scanf("%d %d %d", &n, &v, &k);
		for (i=0; i<n; i++)
			scanf("%d", &value[i]);
		for (i=0; i<n; i++)
			scanf("%d", &volume[i]);
		memset(dp, 0, sizeof(dp));
		for (i=0; i<n; i++)
		{
			for (j=v; j>=volume[i]; j--)
			{
				for (l=0; l<k; l++)
				{
					a[l] = dp[j-volume[i]][l] + value[i];
					b[l] = dp[j][l];
				}
				a[k] = b[k] = -1;
				for (x=y=z=0; (a[x]!=-1 || b[y]!=-1) && z<k; )
				{
					if (a[x]>b[y])
					{
						dp[j][z] = a[x];
						x++;
					}
					else
					{
						dp[j][z] = b[y];
						y++;
					}
					if (dp[j][z]!=dp[j][z-1])
						z++;
				}
			}
		}
		printf("%d\n", dp[v][k-1]);
	}
	return 0;
}