hdu 2639Bone Collector II 01背包第k大问题

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>

using namespace std;

int n;
int N, V, K;
int dp[1005][105];
int v[105], w[105];
int sum[305];
int main()
{
    scanf("%d", &n);
    while(n--){

        scanf("%d%d%d", &N, &V, &K);

        for(int i = 0; i < N; i++) {

            scanf("%d", &v[i]);
        }
        for(int i = 0; i <N; i++) {

            scanf("%d", &w[i]);
        }
        memset(dp, 0, sizeof(dp));

        for(int i = 0; i < N; i++) {


            for(int j = V; j >= w[i]; j--) {

                int ans = 0;
                for(int t = 1; t <= K; t++) {  

                    sum[ans++] = dp[j][t];          //大小为j的背包空间，的价值；
                    sum[ans++] = dp[j - w[i]][t] + v[i];  //放入底i个物品后， 所能产生的价值；
                }
                sort(sum, sum+ans);
                int pos = 1;
                for(int k = ans-1; k >= 0; k--) {

                    if(K < pos)        //第K大数K小于1时跳出，直接输出结果；
                        break;
                    if(k == ans -1 || sum[k] != sum[k+1]) //除去重复的数，
                        dp[j][pos++] = sum[k];
                }
            }
        }
        printf("%d\n", dp[V][K]);
    }
    return 0;
}