容易发现只有每一列有多少行被访问有用,考虑贪心。枚举最右边在哪一列,为到达该行,左边每一列最少选 1 1 1 行,最多选 n n n 行,贪心选最大的,剩下的加到最大的哪一行区即可。实现时,每次加入一列,就将一整列计入答案并加入堆,再从小根堆中不断弹出最小的数,直至剩下 k k k 个。注意要选前几大的 n n n 行,所以可以从左到右枚举,用堆维护。当前所有值,支持每次查询前 k k k 大以及它们的和。
对于一整列计入答案,按数组输入即可。
#include#include #include #include #define LL long long using namespace std; LL ans,n,m,k,size,sum,maxn; LL f[1000100]; priority_queue , greater > 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; }



