hdu2639 Bone Collector II（01背包）

题目链接：

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

题目大意：

求出背包里面第k大的价值。

思路：

如果是求最大价值，可以直接利用01背包求得，此时状态方程为dp[j]=max(dp[j],dp[j-w[i]]+v[i])。

现在是求第k大的，显然不能直接套用了，但还是利用背包，只是这时就要加一维来限制条件了。

我可以设一个dp[j][k]，表示物品容量为 j 时第k大的价值。

这时就有两种状态：dp[j[k]和dp[ j-w [i] ] [k]+v[i]。

然后我就可以开数组将这两个状态存起来，最后通过比较获得前k优的解。

代码：

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
int dp[1005][35],d1[1005],d2[1005];
int max(int a,int b)
{
  if(a>b)return a;
  else return b;	
}
int main()
{
	int T,N,V,K,i,j,k,a[105],v[105];
	scanf("%d",&T);
	while(T--)
	{
		memset(d1,0,sizeof(d1));
		memset(d2,0,sizeof(d2));
		memset(v,0,sizeof(v));
		memset(a,0,sizeof(a));
		scanf("%d%d%d",&N,&V,&K);
		for(i=1;i<=N;i++)
		scanf("%d",&v[i]);
		for(i=1;i<=N;i++)
		scanf("%d",&a[i]);
     	memset(dp,0,sizeof(dp));
	   for(i=1;i<=N;i++)
	   {
   		for(j=V;j>=a[i];j--)
   		{
		   	for(k=1;k<=K;k++)
		   	{
	   			d1[k]=dp[j][k];
	   			d2[k]=dp[j-a[i]][k]+v[i];    //将两种状态的前k优解全部记录下来。
	   		}
	   		int q=1;
	   		int x,y;
	   		x=y=1;
	   		d1[K+1]=d2[K+1]=-1;           //防止其中一个全部用完了，而前k优个解还没有全部获得
	   		while(q<=K&&(x<=K||y<=K))    //取最优的解
	   		{
		   		if(d1[x]>d2[y])
		   		{
		   			dp[j][q]=d1[x];
		   			x++;
		
		   		}
		   		else if(d1[x]<=d2[y]){
		   			dp[j][q]=d2[y];
		   			y++;
		   		}
	
 		   		if(dp[j][q]!=dp[j][q-1])q++;      //去重。
		   	}
		   }
   	}
   	printf("%d\n",dp[V][K]);
	}
	return 0;
	return 0;
}