思路:从左到右依次遍历
每一次根据贪心,每一次加入n个数,如果加入的数超过n个,那么就将最小的元素弹出n-1个(因为要保证1~x中每一列至少1个在队内
记录弹出数的最大值,好让空隙填补
Code
#include<bits/stdc++.h>
#define re register
#define inl inline
#define int long long
using namespace std;
const int N=1e6+10;
int read(){
int sum=0,f=1;char c=getchar();
while(!isdigit(c)){if(c=='-') f=-1;c=getchar();}
while(isdigit(c)){sum=(sum<<3)+(sum<<1)+(c^48);c=getchar();}
return sum*f;
}
int n,m,k;
int a[N];
priority_queue<int,vector<int>,greater<int> > q;
signed main(){
n=read(),m=read(),k=read();
for(re int i=1;i<=m;i++) a[i]=read();
int tmp=0,ans=0;
int sum=0;int maxn=0;
for(re int i=1;i<=m;i++){
if(n==1){
if(i>k){
printf("%lld",sum);
return 0;
}
else{
sum+=a[i];
continue;
}
}
q.push(a[i]);
tmp+=n,sum+=a[i]*n;
while(tmp>k&&!q.empty()){
int res=q.top();
tmp-=n-1;
sum-=(n-1)*q.top();
maxn=max(maxn,q.top());
q.pop();
}
ans=max(ans,sum+min(n-1,k-tmp)*maxn);
if(i>=k) break;
}
printf("%lld",ans);
}
总结:感觉堆的题目大部分都是关于贪心的