hdu 2639 Bone Collector II

求第k大的揹包

每個容量v都對應k個不同的價值，當考慮放第i件物品，容量爲j時，價值爲第k大時，我們要麼不放，要麼放，那麼每個k就對應2個情況，dp[j][k]和dp[j-w[i]][k],總共有2×K種不同價值，那麼我們排好序後，取前K大且兩兩不同的價值，存入dp[j][k]中，更新了考慮第i個揹包的值

這裏其實有更好的實現方法，具體可以參見http://hi.baidu.com/chenyun00/blog/item/f1c7ae545e9e2a491138c26b.html

因爲我們遍歷k時，dp[j][k]和dp[j-w[i]][k]都是有序遍歷的，即如果用兩個數組分開來表示的話，我們用兩個指針就能在O(K)時間內找到前K大且兩兩不同的值

hdu上re居然報wa，囧

代碼：

#include<iostream>
#include<memory.h>
#include<string>
#include<cstdio>
#include<algorithm>
#include<math.h>
#include<stack>
#include<queue>
#include<vector>
#include<map>
using namespace std;
int dp[1005][70],w[1005],v[1005],tmp[105];
int a[50],b[50];
int main()
{
	int n,V,K,i,j,k,T,cnt;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d",&n,&V,&K);
		memset(dp,0,sizeof(dp));
		for(i=1;i<=n;i++)
		{
			scanf("%d",&v[i]);
		}
		for(i=1;i<=n;i++)
		{
			scanf("%d",&w[i]);
		}
		for(i=1;i<=n;i++)
		{
			for(j=V;j>=w[i];j--)
			{
				cnt=0;
				//memset(tmp,0,sizeof(tmp));
				for(k=1;k<=K;k++)
				{
					tmp[cnt++]=dp[j][k];
					tmp[cnt++]=dp[j-w[i]][k]+v[i];
				}
				sort(tmp,tmp+cnt);
				int x=1;
				for(k=cnt-1;k>=0;k--)
				{
					if(x>K)//2*K>=60,開35會wa,需要break，不特判的話要開到61以上 
						break;
					if(k==cnt-1||tmp[k]!=tmp[k+1])
						dp[j][x++]=tmp[k];
				}
			}
		}
		printf("%d\n",dp[V][K]);
	}
	return 0;
}