Codeforces Round 894 (Div. 3) E. Kolya and Movie Theatre

对于所选择的序列$a_1,a_2,a_3,a_4..a_k$，我们最后只需要减去$k\ast d$就可以了

我们可以枚举最后一个数字，根据优先队列来维护前面所有数字里，最大的$m-1$个数字值之和

具体如何维护呢？

构建最小堆，只要堆的容量到了m，那就剔除该最小的数字$x$，同时将$sum-=x$

#include 
typedef long long ll;
using namespace std;
void solve()
{
    ll n, m, d;
    cin >> n >> m >> d;
    vector<int> v(n + 1);
    for(int i = 1; i <= n; i++)
    {
        cin >> v[i];
    }
    priority_queue<int, vector<int>, greater<int>> q;
    ll sum = 0;
    ll res = 0;
    for(int i = 1; i <= n; i++)
    {
        if(v[i] >= 0)
        {
            sum += v[i];
            //超出容量 丢掉最小的
            if(q.size() == m)
            {
                sum -= q.top();
                q.pop();
            }
            q.push(v[i]);
            res = max(res, sum - i * d);
        }
    }
    cout << res << "\n";
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t;
    cin >> t;
    while (t--)
    {
        solve();
    }
}