选数游戏（ybtoj-二叉堆）

题目描述

解析

一道很考验代码能力与思维的题
（我不是在为自己的菜找理由）
首先由于可以每一列都有类似于环的性质
所以我们可以忽略点的出入的纵坐标的位置，只考虑每列选几个
首先，每个列都得至少选一个（这样才能走过去），然后剩下的任意选择，个数不超过k-i即可
这样我们就有一个大概的贪心思路了
从前往后枚举选点结束的位置(也就是最右列的位置），不断尝试更新ans
维护一个优先队列，每当往右拓展一列时，至少进队一个（且它以后不能出列）
然后如果队首的元素比当前列的少就一直弹出并换成当前列的
时间复杂度：mlogm

注意：k也要开long long！

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const int N=1e6+100;
ll n,m,k;
ll a[N]; 
ll now,ans=0,tot;
struct node{
	ll x,num;
	bool operator < (const node y)const{return x>y.x;}
};
priority_queue<node>q;
int main(){
	scanf("%lld%lld%lld",&n,&m,&k);
	int flag=-1;
	for(int i=1;i<=m;i++){
		scanf("%lld",&a[i]);
		int flag=1;
		while(!q.empty()&&tot>k-n){
			node o=q.top();q.pop();
			if(o.x>a[i]&&!flag){
				//cout<<"ok\n";
				q.push(o);
				break;
			}
			flag=0;
//			printf("i=%d tot=%d now=%d num=%d\n",i,tot,o.x,o.num);
			if(o.x>a[i]){
				o.num--;
				tot--;
				now-=o.x;
				q.push(o);
				break;
			}
			if(o.num<=tot-(k-n)){
				now-=o.x*o.num;
				tot-=o.num;
			}
			else{
				ll jian=tot-(k-n);
				o.num-=jian;
				now-=o.x*jian;
				q.push(o);
				tot-=jian;
			}
		}
		now+=a[i]*min((ll)n,k-tot);
		q.push((node){a[i],min((ll)n,k-tot)-1});
		tot+=min((ll)n,k-tot);
		//printf("i=%d tot=%lld now=%lld\n",i,tot,now);
		ans=max(ans,now);
		if(k<=i) break;
	}
	printf("%lld",ans);
	return 0;
}
/*
4 6 10
2 3 5 1 8 1
*/

thanks for reading！