2023南京G - Knapsack

2023南京

//按价值从大到小排序
//1背包得到dp表 dp【i】【j】表示 i之后  个占用空间为j的最大价值
//2枚举一个分界点m 前m个取k个最大的  后面的取dp最大的
//3前m个维护一个“堆” 优先队列 小根堆 每次弹出最小的


#include <bits/stdc++.h>
#define ll long long
using namespace std;
ll dp[5005][10004],mxdp[5006];
struct code{
    ll a;
    ll b;
}wp[5005];

bool cmp2(code x,code y){
    return x.a>y.a;
}
priority_queue<int, vector<int>, greater<int> > q;
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    ll n,w,k;
    cin >> n >> w >> k;
    for(int i = 1;i <= n;i++){
        cin >> wp[i].a >> wp[i].b;
    }
    sort(wp+1,wp+n+1,cmp2);

    for(int i = n;i >= 1;i--){
        for(int j = 1;j <= w;j++){
            if(j-wp[i].a >= 0)
                dp[i][j] = max(dp[i+1][j] , dp[i+1][j-wp[i].a] + wp[i].b);
            else 
                dp[i][j] = dp[i+1][j];

            mxdp[i] = max(mxdp[i],dp[i][j]);    //i以后中最大的dp
        }
        // cout << i <<" " << mxdp[i] << "\n";
    }


    ll num = 0,ans = 0;
    for(int m = 0;m <= n;m++){
        if(m!=0)q.push(wp[m].b);
        if(m > k){
            num -= q.top();
            q.pop();
        }
        num += wp[m].b;
        ans = max(ans,mxdp[m+1]+num);
    }
    cout << ans << "\n";
    return 0;
}