codeforces 贪心 Traps

题目描述：

 题目大意：给定一个数组，从左到右你必须跨越这些数，代价是这个数的值，你也可以选择  次跳过某个数，这样该数代价为0，但之后的每个数代价都会加1， 问最多 k 次跳跃的机会，求最小的代价。

算法思想：贪心

        首先要明确的一点是，这 k 次跳跃必须全部用完才是最优的。如果我们只用 k - 1次跳跃， 那么最靠右我们不跳跃的地方如果我们跳跃过去，结果一定比 k - 1次跳跃效果更好。

        明确了这一点后 ，考虑第一个跳跃的点 i , 首先我们会减少代价 a[i] ，然后此后还有 n - i 个位置代价会加 1, 所以我们增加代价  n - i,  又因为我们还有 k - 1次跳跃的机会，那么这 n - i 多加的代价中有 k - 1 个是无效的。所以这一点增益 a[i] - (n - i ) + k - 1.

        我们发现，通过累加， - n 和 k - 1这两项都可以提出来，对于总和，这是个定值。而 i 处的增益我们可以看成是 a[i] + i。最后我们排序从大到小取出最大的 k 个就行了。 

AC代码：

 

#include <bits/stdc++.h>
using namespace std;

#define all(x) (x).begin(), (x).end()

void solve() {
    int n, k;
    cin >> n >> k;
    long long ans = 0;
    vector<int> a(n);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
        ans += a[i];
        a[i] += i + 1;
    }
    sort(all(a));
    reverse(all(a));
    for (int i = 0; i < k; i++) ans -= a[i];
    for (int i = 0; i < k; i++) {
        ans += n;
        ans -= i;
    }
    cout << ans << "\n";
}

int main() {
    int t;
    cin >> t;
    while (t--) solve();
    return 0;
}