原题链接
题解
alice只有两种决策
- 一个也不选
- 选 \(k\) 个以上
如果选 \(k\) 个以上,bob肯定会使其中 \(b_i\) 最大的 \(k\) 个免费,所以我们干脆把 \(b\) 降序排序,然后遍历第 \(k\) 小的 \(i\),由于这 \(k\) 个数无论怎么选都是赔,所以我们我们选 \([1,i]\) 里前 \(k\) 小的 \(a_i\),之后就贪心地选 \(b_i > a_i\) 的
code
#include<bits/stdc++.h>
using namespace std;#define ll long longstruct node
{ll a,b;
}c[200005];
bool cmp(node x,node y)
{return x.b>y.b;
}ll suf[200005]={0};
int main()
{ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);ll t;cin>>t;while(t--){ll n,k;cin>>n>>k;for(ll i=1;i<=n;i++) cin>>c[i].a;for(ll i=1;i<=n;i++) cin>>c[i].b;sort(c+1,c+1+n,cmp);ll sum=0;priority_queue<ll> q;for(ll i=1;i<=k;i++){sum+=c[i].a;q.push(c[i].a);}suf[n+1]=0;for(ll i=n;i>k;i--){suf[i]=suf[i+1];if(c[i].b>c[i].a) suf[i]+=c[i].b-c[i].a;}if(!k)//针对堆,特判一下{cout<<suf[1]<<'\n';continue;}ll ans=max(0LL,suf[k+1]-sum);for(ll i=k+1;i<=n;i++){ll big=q.top();if(c[i].a<big){sum-=big;sum+=c[i].a;q.pop();q.push(c[i].a);}ans=max(ans,suf[i+1]-sum);}cout<<ans<<'\n';}return 0;
}