D. Shop Game

Problem - 1969D - Codeforces  

观察分析A和B的优先选取性质，B拿价格高，A可选择买不买，A需让出k个给B白嫖；故将bi逆序排序，线性枚举，利用优先队列求得前 i 项中 k 个最小的 ai 的和给B白嫖，i+1~n的后缀和为纯利润；巧用基础工具；

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int max_n=2e5+5;
int n,k,suf[max_n];
struct Node{int a,b;bool operator <(const Node &x)const{return b>x.b;}}node[max_n];
void solve()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)cin>>node[i].a;
    for(int i=1;i<=n;i++)cin>>node[i].b;
    for(int i=0;i<=n+1;i++)suf[i]=0;
 
    sort(node+1,node+1+n);
 
    for(int i=n;i>=1;i--){
        suf[i]+=suf[i+1]+max(0LL,node[i].b-node[i].a);
    }
 
    if(k==0){
        cout<<suf[1]<<endl;
        return;
    }
 
    priority_queue<int>q;
 
    int cnt=0,cost=0,ans=0,mx;
    for(int i=1;i<=n;i++){
        if(cnt<k){
            if(node[i].a<node[i].b){
            q.push(node[i].a);
            cost+=node[i].a;
            mx=q.top();
            cnt++;
            }
        }else{
            ans=max(ans,suf[i]-cost);
            if(node[i].a<mx){
                cost-=mx;
                cost+=node[i].a;
                q.pop();
                q.push(node[i].a);
                mx=q.top();
            }
        }
    }
 
    cout<<ans<<endl;
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
 
    int t;cin>>t;
    while(t--){
        solve();
    }
 
    return 0;
}