首先，应当选 $happiness$ 中最大的 $k$ 的数。

这些数要按照什么顺序选呢？

由于小的数减成 $0$ 就不再减少了，优先选大的更好。

比如 $2,1,1，$如果按照 $1,1,2$的顺序选，答案为 $1+0+0=1$。但按照 $2,1,1$的顺序选，答案为 $2+0+0=2$更优。

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n, k;
    cin >> n >> k;
    vector<int> happiness(n);
    for (int i = 0; i < n; i++) {
        cin >> happiness[i]; 
    }
    long long ans = 0;
    sort(happiness.begin(), happiness.end(), greater<int>()); 
    for (int i = 0; i < k && happiness[i] > i; i++) {
            ans += happiness[i] - i;
        }

    cout << ans;
    return 0;
}