CF1006B 题解
水。
Q:首先,如果不考虑最大,怎么办?
A:我们就把 k k k 个最大的数分别放在 k k k 个序列中,就行了。
Q:那么这 k k k 个数位置都是不相同的,那么不就是每一个序列带着一些比它小的而已吗?
A:
好的被膜拜的人不是我!(逃
所以,我们只要将这个序列排序,找 k k k 大的数,记录它们的位置。设 a i a_i ai 为它从 1 1 1 开始标号的位置。
然后,对于 1 < i < k 1 < i < k 1<i<k,我们都可以得出,长度为 a i − a i − 1 a_i - a_{i - 1} ai−ai−1。
对于 i = 1 i = 1 i=1 或者 i = n i = n i=n,我们特判一下(毕竟最后一个不一定是 n n n,第一个不一定是 1 1 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;
}