hdu2639

链接:点击打开链接

题意:求01背包的第K大值

代码:

<span style="font-family:FangSong_GB2312;font-size:18px;">#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int v[105],w[105],dp[1005][35],tmp[2005];
int main(){
    int N,K,V,i,j,k,t,l,cnt;
    scanf("%d",&t);
    while(t--){                                 //dp[i][j]代表容量是i第j大
        scanf("%d%d%d",&N,&V,&K);
        for(i=0;i<N;i++)
        scanf("%d",&w[i]);
        for(i=0;i<N;i++)
        scanf("%d",&v[i]);
        memset(dp,0,sizeof(dp));
        for(i=0;i<N;i++){
            for(j=V;j>=v[i];j--){
                cnt=0;
                for(k=1;k<=K;k++){
                    tmp[cnt++]=dp[j-v[i]][k]+w[i];
                    tmp[cnt++]=dp[j][k];
                }                               //把所有情况存起来
                sort(tmp,tmp+cnt,greater<int>());
                l=unique(tmp,tmp+cnt)-tmp;
                for(k=1;k<=min(l,K);k++)        //去重排序赋值
                dp[j][k]=tmp[k-1];
            }
        }
        printf("%d\n",dp[V][K]);
    }
    return 0;
}
</span>

你可能感兴趣的:(hdu2639)