【数学】Kolya and Movie Theatre—CF1862E

Kolya and Movie Theatre—CF1862E

• 遇到数学式子，想不到思路的时候试着将式子化简！

翻译

最近，Kolya发现他所在的城市即将有一家新电影院开业，将连续 $n$ 天每天上映一部新电影。因此，第 $1\le i\le n$ 天，电影院将上映第 $i$ 部电影的首映。此外，Kolya还了解到了电影的排片表，并为每部电影分配了娱乐价值，用 $a_i$ 表示。

然而，Kolya越长时间不去电影院，下一部电影的娱乐价值减少越多。这个减少等于 $d\cdot cnt$，其中 $d$ 是预定的值，$cnt$ 是自上次去电影院以来的天数。还知道 Kolya 在新电影院开业前一天成功参观了另一家电影院，即第 $0$ 天。因此，如果我们第一次在第 $i$ 天参观电影院，则自上次去电影院以来的天数 $cnt$ 将等于 $i$。

例如，如果 $d=2$，$a=[3,2,5,4,6]$，那么通过参观第 $1$ 和第 $3$ 部电影，第 $1$ 天的 $cnt$ 值将等于 $1-0=1$，第 $3$ 天的 $cnt$ 值将等于 $3-1=2$，因此所有电影的总娱乐价值将为 $a_1-d\cdot 1+a_3-d\cdot 2=3-2\cdot 1+5-2\cdot 2=2$。

不幸的是，Kolya只有时间参观至多 $m$ 部电影。帮助他制定一个计划，以使他参观的所有电影的总娱乐价值最大化。

输入

每个测试包含多个测试用例。第一行包含一个整数 $t$ ($1\le t\le 10^4$) —— 测试用例的数量。接下来是每个测试用例的描述。

每个测试用例的第一行包含三个整数 $n$，$m$，$d$ ($1\le n\le 2\cdot 10^5$，$1\le m\le n$，$1\le d\le 10^9$)。

每组输入数据的第二行包含 $n$ 个整数 $a_1,a_2,\dots ,a_n$ ($-10^9\le a_i\le 10^9$) —— 电影的娱乐价值。

保证所有测试用例中 $n$ 的总和不超过 $2\cdot 10^5$。

输出

对于每个测试用例，输出一个整数 —— Kolya 可以获得的最大总娱乐价值。

注意

第一个测试用例在问题陈述中有解释。

在第二个测试用例中，最优解是不去参观任何电影。

在第三个测试用例中，最优解是参观第 $2$、$3$、$5$ 和 $6$ 部电影，因此所参观电影的总娱乐价值将为 $45-6\cdot 2+1-6\cdot 1+39-6\cdot 2+11-6\cdot 1=60$。

思路

这个题主要是数学化简，再加上一点点数据结构。

如果我们在数字为 $i_1,i_2,\dots ,i_k$ 的日子去电影院，所看电影的总娱乐价值将等于 $(a_{i_1}-d\cdot i_1)+(a_{i_2}-d\cdot (i_2-i_1))+\dots +(a_{i_k}-d\cdot (i_k-i_{k-1}))=(a_{i_1}+a_{i_2}+\dots +a_{i_k})-d\cdot i_k$。

因此，我们只需遍历 Kolya 最后一次去电影院的日子 $i_k$，并在前缀 $[1,i_k]$ 上维护最大的 $m$ 个非负元素。例如，可以使用优先队列来实现。解决方案的总复杂度将为 $O(n\log n)$。

$Code$

#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;

int n, m, d;
int a[N];

void solve() {
	cin >> n >> m >> d;
	for (int i = 1; i <= n; i ++) {
		cin >> a[i];
	}
	
	priority_queue <int, vector<int>, greater<int>> q; 
	// 维护最大的m个>0的元素
	
	int res = 0;
	int sum = 0; // 维护的元素的和
	for (int last = 1; last <= n; last ++) { // 遍历选中的最后一个元素的下标
		if (a[last] > 0) {
			sum += a[last];
			q.push(a[last]);
			if (sz(q) > m) {
				sum -= q.top();
				q.pop();
			}
			res = max(res, sum - d * last);
		}
	}
	
	cout << "        ";
	cout << res << "\n";
}

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int T = 1;
	cin >> T; cin.get();
	while (T --) solve();
	return 0;
}