HDU-2639-Bone Collector II

这个题是HDU-2602-Bone Collector升级版,改为求第k大的值

思路:

将背包的维数增加一维,存的时候注意去重~

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=1001;
int n,m,k,dp[maxn][31],v[maxn],p[maxn];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
	memset(dp,0,sizeof(dp));
	scanf("%d%d%d",&n,&m,&k);
	for(int i=0;i<n;i++)
	    scanf("%d",&p[i]);
	for(int i=0;i<n;i++)
	    scanf("%d",&v[i]);
	int ans=0;
	for(int i=0;i<n;i++)
	    for(int j=m;j>=v[i];j--)
	    {
		int ita[32],itb[32];
		for(int l=0;l<k;l++)
		{
		    ita[l]=dp[j][l];
		    itb[l]=dp[j-v[i]][l]+p[i];
		}
		ita[k]=itb[k]=-1;
		int cura=0,curb=0;
		for(int l=0;(ita[cura]!=-1||itb[curb]!=-1)&&l<k;l++)
		{
		    if(ita[cura]>itb[curb])
			dp[j][l]=ita[cura++];
		    else
			dp[j][l]=itb[curb++];
		    if(l>0&&dp[j][l]==dp[j][l-1])
			l--;
		}
	    }
	printf("%d\n",dp[m][k-1]);
    }
    return 0;
}


你可能感兴趣的:(背包)