Kolya and Movie Theatre—CF1862E
- 遇到数学式子,想不到思路的时候试着将式子化简!
翻译
最近,Kolya发现他所在的城市即将有一家新电影院开业,将连续 n n n 天每天上映一部新电影。因此,第 1 ≤ i ≤ n 1 \le i \le n 1≤i≤n 天,电影院将上映第 i i i 部电影的首映。此外,Kolya还了解到了电影的排片表,并为每部电影分配了娱乐价值,用 a i a_i ai 表示。
然而,Kolya越长时间不去电影院,下一部电影的娱乐价值减少越多。这个减少等于 d ⋅ c n t d \cdot cnt d⋅cnt,其中 d d d 是预定的值, c n t cnt cnt 是自上次去电影院以来的天数。还知道 Kolya 在新电影院开业前一天成功参观了另一家电影院,即第 0 0 0 天。因此,如果我们第一次在第 i i i 天参观电影院,则自上次去电影院以来的天数 c n t cnt cnt 将等于 i i i。
例如,如果 d = 2 d=2 d=2, a = [ 3 , 2 , 5 , 4 , 6 ] a=[3, 2, 5, 4, 6] a=[3,2,5,4,6],那么通过参观第 1 1 1 和第 3 3 3 部电影,第 1 1 1 天的 c n t cnt cnt 值将等于 1 − 0 = 1 1-0=1 1−0=1,第 3 3 3 天的 c n t cnt cnt 值将等于 3 − 1 = 2 3-1=2 3−1=2,因此所有电影的总娱乐价值将为 a 1 − d ⋅ 1 + a 3 − d ⋅ 2 = 3 − 2 ⋅ 1 + 5 − 2 ⋅ 2 = 2 a_1 - d \cdot 1 + a_3 - d \cdot 2 = 3 - 2 \cdot 1 + 5 - 2 \cdot 2 = 2 a1−d⋅1+a3−d⋅2=3−2⋅1+5−2⋅2=2。
不幸的是,Kolya只有时间参观至多 m m m 部电影。帮助他制定一个计划,以使他参观的所有电影的总娱乐价值最大化。
输入
每个测试包含多个测试用例。第一行包含一个整数 t t t ( 1 ≤ t ≤ 1 0 4 1 \le t \le 10^4 1≤t≤104) —— 测试用例的数量。接下来是每个测试用例的描述。
每个测试用例的第一行包含三个整数 n n n, m m m, d d d ( 1 ≤ n ≤ 2 ⋅ 1 0 5 1 \le n \le 2 \cdot 10^5 1≤n≤2⋅105, 1 ≤ m ≤ n 1 \le m \le n 1≤m≤n, 1 ≤ d ≤ 1 0 9 1 \le d \le 10^9 1≤d≤109)。
每组输入数据的第二行包含 n n n 个整数 a 1 , a 2 , … , a n a_1, a_2, \ldots, a_n a1,a2,…,an ( − 1 0 9 ≤ a i ≤ 1 0 9 -10^9 \le a_i \le 10^9 −109≤ai≤109) —— 电影的娱乐价值。
保证所有测试用例中 n n n 的总和不超过 2 ⋅ 1 0 5 2 \cdot 10^5 2⋅105。
输出
对于每个测试用例,输出一个整数 —— Kolya 可以获得的最大总娱乐价值。
注意
第一个测试用例在问题陈述中有解释。
在第二个测试用例中,最优解是不去参观任何电影。
在第三个测试用例中,最优解是参观第 2 2 2、 3 3 3、 5 5 5 和 6 6 6 部电影,因此所参观电影的总娱乐价值将为 45 − 6 ⋅ 2 + 1 − 6 ⋅ 1 + 39 − 6 ⋅ 2 + 11 − 6 ⋅ 1 = 60 45 - 6 \cdot 2 + 1 - 6 \cdot 1 + 39 - 6 \cdot 2 + 11 - 6 \cdot 1 = 60 45−6⋅2+1−6⋅1+39−6⋅2+11−6⋅1=60。
思路
这个题主要是数学化简,再加上一点点数据结构。
如果我们在数字为 i 1 , i 2 , … , i k i_1, i_2, \ldots, i_k i1,i2,…,ik 的日子去电影院,所看电影的总娱乐价值将等于 ( a i 1 − d ⋅ i 1 ) + ( a i 2 − d ⋅ ( i 2 − i 1 ) ) + … + ( a i k − d ⋅ ( i k − i k − 1 ) ) = ( a i 1 + a i 2 + … + a i k ) − d ⋅ i k (a_{i_1}-d \cdot i_1)+(a_{i_2}-d \cdot (i_2 - i_1))+\ldots+(a_{i_k}-d \cdot (i_k-i_{k-1}))=(a_{i_1}+a_{i_2}+\ldots+a_{i_k})-d \cdot i_k (ai1−d⋅i1)+(ai2−d⋅(i2−i1))+…+(aik−d⋅(ik−ik−1))=(ai1+ai2+…+aik)−d⋅ik。
因此,我们只需遍历 Kolya 最后一次去电影院的日子 i k i_k ik,并在前缀 [ 1 , i k ] [1, i_k] [1,ik] 上维护最大的 m m m 个非负元素。例如,可以使用优先队列来实现。解决方案的总复杂度将为 O ( n log n ) O(n \log{n}) O(nlogn)。
C o d e Code 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;
}