CF1006B 题解

CF1006B 题解

水。

Q：首先，如果不考虑最大，怎么办？

A：我们就把 $k$ 个最大的数分别放在 $k$ 个序列中，就行了。

Q：那么这 $k$ 个数位置都是不相同的，那么不就是每一个序列带着一些比它小的而已吗？

A：
好的被膜拜的人不是我！（逃

所以，我们只要将这个序列排序，找 $k$ 大的数，记录它们的位置。设 $a_i$ 为它从 $1$ 开始标号的位置。

然后，对于 $1<i<k$，我们都可以得出，长度为 $a_i-a_{i-1}$。

对于 $i=1$ 或者 $i=n$，我们特判一下（毕竟最后一个不一定是 $n$，第一个不一定是 $1$）

最后上我的丑陋代码：

#include <bits/stdc++.h>
#define int long long
using namespace std;
signed main(){
    int n, k;
    cin >> n >> k;
    pair<int, int> a[n];
    for(int i = 0;i < n;i++){
        cin >> a[i].first;
        a[i].second = i;
    }
    sort(a, a + n, greater<pair<int, int> >());
    vector<int> ans;
    int sum = 0;
    for(int i = 0;i < k;i++){
        sum += a[i].first;
        ans.push_back(a[i].second);
    }
    cout << sum << endl;
    sort(ans.begin(), ans.end());
    if(ans.size() == 1)
        cout << n << endl;
    else
        cout << ans[0] + 1 << " ";
    for(int i = 1;i < ans.size() - 1;i++){
        cout << ans[i] - ans[i - 1] << " ";
    }
    if(ans.size() > 1)
        cout << ans[ans.size() - 1] - ans[ans.size() - 2] + n - ans[ans.size() - 1] - 1 << endl;
    return 0;
}