HDU 2639 Bone Collector II

还能有比我这更暴力的做法么，直接拿set来维护前k大值……这题的话不能拿一维的来做是因为每次转移会覆盖掉原来的状态，但是这些状态可能会构成新的状态，这些新状态虽然不是最优但是是前k大值，而一维可能会丢失这部分。

#include <algorithm>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

#include <map>
#include <set>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <bitset>
#include <memory>
#include <complex>
#include <numeric>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include <ctype.h>
#include <locale.h>

using namespace std;

#pragma pack(4)

const double  eps = 1e-8;
const double   pi = acos(-1.0);
const int     inf = 0x7f7f7f7f;

#define loop(a,n)                            \
    for(int i=0;n>i;i++)                     \
        cout<<a[i]<<(i!=n-1?' ':'\n')
#define loop2(a,n,m)                         \
    for(int i=0;n>i;i++)                     \
        for(int j=0;m>j;j++)                 \
            cout<<a[i][j]<<(j!=m-1?' ':'\n')

#define   at(a,i) ((a)&(1<<(i)))
#define   nt(a,i) ((a)^(1<<(i)))
#define set1(a,i) ((a)|(1<<(i)))
#define set0(a,i) ((a)&(~(1<<(i))))

#define cmp(a,b) (fabs((a)-(b))<eps?0:(((a)-(b))>eps?+1:-1))

#define lmax(a,b) ((a)>(b)?(a):(b))
#define lmin(a,b) ((a)<(b)?(a):(b))
#define fmax(a,b) (cmp(a,b)>0?(a):(b))
#define fmin(a,b) (cmp(a,b)<0?(a):(b))

#define ML int tcase,times=1;for(cin>>tcase;tcase>=times;times++)

const int MAXV = 1002;

set<int> dp[MAXV];
int n,v,K,a[MAXV],b[MAXV];

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("Bone Collector II.txt","r",stdin);
    #else
    #endif

    ML
    {
        scanf("%d %d %d",&n,&v,&K);
        for(int i=0;n>i;i++)
        {
            scanf("%d",&b[i]);
        }
        for(int i=0;n>i;i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i=0;MAXV>i;i++) dp[i].clear();
        dp[0].insert(0);
        set<int> ans;
        ans.insert(0);
        for(int i=0;n>i;i++)
        {
            for(int j=v;j-a[i]>=0;j--)
            {
                for(set<int>::iterator k=dp[j-a[i]].begin();k!=dp[j-a[i]].end();k++)
                {
                    dp[j].insert((*k)+b[i]);
                    ans.insert((*k)+b[i]);
                    while(K<dp[j].size()) dp[j].erase(dp[j].begin());
                    while(K<ans.size()) ans.erase(ans.begin());
                }
            }
        }
        if(ans.size()<K) printf("0\n");
        else
        {
            set<int>::iterator now=ans.begin();
            for(int i=0;ans.size()!=K+i;i++) now++;
            printf("%d\n",*now);
        }
    }

    return 0;
}