Codeforces ~ 1006B ~ Polycarp's Practice（贪心 + 模拟）

题意

将一个长度为n的数组划分为k段，每段的最大值加起来的和最大为多少，怎么划分输出每段多长（任意解）？

---

思路

选最大的k个数字相加就是答案，怎么划分，我们可以把他们的下标存一下，然后根据下标求出段长就OK了。

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2005;
int n, k;
pair<int, int> a[MAXN];
int main()
{
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i++) scanf("%d", &a[i].first), a[i].second = i;
    sort(a, a+n);
    int ans = 0;
    vector<int> v;
    v.push_back(-1);
    for (int i = n-1; i >= n-k; i--)
    {
        ans += a[i].first;
        v.push_back(a[i].second);
    }
    printf("%d\n", ans);
    sort(v.begin(), v.end());
    v[v.size()-1] = n-1;
    for (int i = 1; i < v.size(); i++)
    {
        printf("%d ", v[i]-v[i-1]);
    }
    return 0;
}
/*
8 3
5 4 2 6 5 1 9 2
*/