Kolya and Movie Theatre--优先队列

Kolya and Movie Theatre - 洛谷

 

#include <bits/stdc++.h>
#define int long long //(有超时风险)
#define PII pair<int,int>
#define endl '\n'
#define all(x) x.begin(), x.end()
#define LL __int128

using namespace std;

const int N=2e6+10,M=1e3+10,mod=998244353,INF=0x3f3f3f3f;

int a[N],b[N],c[N],pre[N];
int g[M][M];

//如果最后一个要用到第i个数的话，那么减去i*d
//我们用优先队列就行了，然后减去当前的i*d；
void solve()
{
    int n,m,d;cin>>n>>m>>d;

    for(int i=1;i<=n;i++)
        cin>>a[i];

    priority_queue<int,vector<int>,greater<int>>q;
    int cnt=0;
    int sum=0,ans=0;
    for(int i=1;i<=m;i++)
        q.push(0);
    for(int i=1;i<=n;i++)
    {
        if(a[i]<=0)
            continue;

        if(q.top()<a[i])
        {
            sum=sum-q.top()+a[i];
            q.pop();
            q.push(a[i]);
        }
        ans=max(ans,sum-i*d);
    }
    cout<<ans<<endl;
}

signed main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int _t=1;cin>>_t;
    while(_t--)
        solve();
    return 0;
}