正题

---

题目大意

$n$个物品，有两个人，每个人有一些喜欢的物品。

选$m$个物品，至少选择$k$个第一个人喜欢的和$k$个第二个人喜欢的物品

---

解题思路

首先我们必定是选最小的
我们从小到大枚举选择多少两个人都喜欢的物品$i$，然后每人选择$k-x$只有这个人喜欢的物品，之后我们将剩下的物品丢进一个数据结构里，然后求前$m-2\ast k+i$个最小的物品。

现在我们需要维护一个数据结构支持每次加入一些物品，然后求前$z$个最小物品的和。我们发现$z$是每次增加一的，可以使用对顶堆，保证第一个堆内只有$z$个物品然后每次如果更优就交换两个堆堆顶的数

时间复杂度$O(n\log n)$

---

$code$

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
const ll N=2e5+10;
priority_queue<ll> q1,q2;
ll n,m,k;
ll ans,cnt,cnt1,cnt2,sum1,sum2,sum,sum0;
ll v[N],a[N],b[N],com[N],p1[N],p2[N];
int main()
{scanf("%lld%lld%lld",&n,&m,&k);for(ll i=1;i<=n;i++)scanf("%lld",&v[i]);ll num;scanf("%lld",&num);for(ll i=1;i<=num;i++){ll x;scanf("%lld",&x);a[x]=1;}scanf("%lld",&num);for(ll i=1;i<=num;i++){ll x;scanf("%lld",&x);b[x]=1;}for(ll i=1;i<=n;i++){if(a[i]&&b[i])com[++cnt]=v[i];else if(a[i])p1[++cnt1]=v[i],sum1+=v[i];else if(b[i])p2[++cnt2]=v[i],sum2+=v[i];else q1.push(-v[i]);}sort(com+1,com+1+cnt);sort(p1+1,p1+1+cnt1);sort(p2+1,p2+1+cnt2);ans=1e18;for(ll i=0;i<=min(cnt,k);i++){sum+=com[i];while(cnt1>k-i){sum1-=p1[cnt1];q1.push(-p1[cnt1]);cnt1--;}while(cnt2>k-i){sum2-=p2[cnt2];q1.push(-p2[cnt2]);cnt2--;}ll ne=m-cnt1-cnt2-i;if(ne<0)continue;if(cnt1+i!=k||cnt2+i!=k)continue;while(!q1.empty()&&q2.size()<ne){q2.push(-q1.top());sum0-=q1.top();q1.pop();}while(!q1.empty()&&!q2.empty()&&-q1.top()<q2.top()){ll z1=-q1.top(),z2=q2.top();q1.pop();q2.pop();sum0+=z1-z2; q1.push(-z2);q2.push(z1);}if(q2.size()>=ne)ans=min(ans,sum0+sum+sum1+sum2);}if(ans==1e18)printf("-1");else printf("%lld",ans);
}