选数游戏(priority_queue)

容易发现只有每一列有多少行被访问有用，考虑贪心。枚举最右边在哪一列，为到达该行，左边每一列最少选 $1$ 行，最多选 $n$ 行，贪心选最大的，剩下的加到最大的哪一行区即可。实现时，每次加入一列，就将一整列计入答案并加入堆，再从小根堆中不断弹出最小的数，直至剩下 $k$ 个。注意要选前几大的 $n$ 行，所以可以从左到右枚举，用堆维护。当前所有值，支持每次查询前 $k$ 大以及它们的和。

对于一整列计入答案，按数组输入即可。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#define LL long long
using namespace std;
LL ans,n,m,k,size,sum,maxn;
LL f[1000100];
priority_queue<LL, vector<LL>, greater<LL> > q;
int main()
{
	scanf("%lld%lld%lld",&n,&m,&k);
	for(LL i=1;i<=min(m,k);i++){
		scanf("%lld",&f[i]);
		q.push(f[i]);
		size+=n,sum+=n*f[i];//将一整列加入堆 
		while(size>k&&q.size()>0){
			LL u=q.top();
			q.pop();
			size-=n-1ll,sum-=(n-1ll)*u;//每列至少留一个数 
			maxn=u;
		}
		ans=max(ans,sum+min(n,k-size)*maxn);//剩下的加到剩下的最大一行内 
	}
	printf("%lld",ans);
	return 0;	
}