题意
给定长度为n的序列,要求分成k段,最大化每段最大值的和
输入
第一行n和k(1 <= k <= n <= 2000)
第二行为序列a[1],a[2],a[3].......
输出
第一行n和k
题解
1.排序排出前kk个数值
2.再把位置从低到高排一下,还原数组
3.碰到前kk个数中其中一个数就分割,分摊到每一组。
代码
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
const int N = 2010;
PII p[N];
bool cmp1(PII a, PII b){return a.first > b.first;}
bool cmp2(PII a, PII b){return a.second < b.second;}
int main()
{
int n, k, sum = 0;
cin >> n >> k;
for(int i = 1; i <= n; i++)
{
cin >> p[i].first;
p[i].second = i;
}
sort(p + 1, p + 1 + n, cmp1);
for(int i = 1; i <= k; i++) sum += p[i].first;
cout << sum << endl;
sort(p + 1, p + 1 + k, cmp2);
if(k == 1)
{
cout << n;
return 0;
}
cout << p[1].second << " ";
for(int i = 2; i < k; i++)
cout << p[i].second - p[i - 1].second << " ";
cout << n - p[k - 1].second;
return 0;
}