正题

题目大意

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

选mmm个物品，至少选择kkk个第一个人喜欢的和kkk个第二个人喜欢的物品

解题思路

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

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

时间复杂度O(nlog⁡n)O(n\log n)O(nlogn)

codecodecode

#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);
}

nssl1477-赛【对顶堆,贪心】相关推荐

NOI.AC-保镖【贪心,对顶堆】
正题题目链接:http://noi.ac/contest/266/problem/795 题目大意 nnn个人第iii个巡逻一次aia_iai秒,休息至少bib_ibi秒. 要求任意时刻都有人 ...
106. 动态中位数【经典 / 对顶堆】
用对顶堆来维护.一个大根堆,一个小根堆. 大根堆存前一半的数据,小根堆存后一半的数据. 需要满足这样的一个条件: 大根堆最大的数<=小根堆最小的数大根堆的数量最多比小根堆的数量多1,这样大根堆 ...
神奇的东西——对顶堆
一.定义就是一个大根堆与一个小根堆的结合,可以用来解决动态区间第k大的问题.可以通过下图来理解. 那么我们如何建一个对顶堆呢? 上文说到,对顶堆是一种基于大根堆和小根堆的数据结构.最开始的时 ...
BZOJ1112洛谷P3466 [POI2008]KLO-Building blocks(砖头)[对顶堆学习笔记]
坑爹的BZOJ因为不让输出方案我WA了若干次据说正解是平衡树/线段树,我不会,就用了个新东西->对顶堆对顶堆一个维护中位数的东西,我们通过维护两个堆,来维所有数的中位数 (显然我们可以排 ...
|Tyvj|NOIP2004|堆|贪心|P1066 合并果子
http://tyvj.cn/p/1066 贪心地先把所有果子放进小根堆,然后取出两个最小值相加后再放进去,在此期间ans+=两个最小值的和,直到只剩1个元素停止 #include<cstdio ...
P7990-[USACO21DEC]Closest Cow Wins S【堆,贪心】
正题题目链接:https://www.luogu.com.cn/problem/P7990 题目大意数轴上有kkk个点是草地,每个草地有不同收益,mmm个点是地方的点,现在你要放置nnn个我方的点 ...
P4989-二进制之谜【堆,贪心】
正题题目链接:https://www.luogu.org/problemnew/show/P4989 题目大意一个二进制数两两配对,要求配对的数不能交叉(用同一个区间但不包含) 0在前1在后要 ...
jzoj1082-合并果子【堆,贪心】
题目有n堆大小不同的果子堆,每次合并两堆消耗它们两堆的重量之和,求把它们全部合并的最小消耗输入输入包括两行,第一行是一个整数n(1<＝n<=10000),表示果子的种类数.第二行包含 ...
【洛谷】【堆+贪心】P1484 种树
[题目描述:] cyrcyr今天在种树,他在一条直线上挖了n个坑.这n个坑都可以种树,但为了保证每一棵树都有充足的养料,cyrcyr不会在相邻的两个坑中种树.而且由于cyrcyr的树种不够,他至多会种 ...

nssl1477-赛【对顶堆,贪心】

正题

题目大意

解题思路

codecodecode

nssl1477-赛【对顶堆,贪心】相关推荐

最新文章

热门文章