题意

给定长度为n的序列,要求分成k段,最大化每段最大值的和

输入

第一行n和k(1 <= k <= n <= 2000)
第二行为序列a[1],a[2],a[3].......

输出

第一行n和k

题解

1.排序排出前k个数值
2.再把位置从低到高排一下,还原数组
3.碰到前k个数中其中一个数就分割,分摊到每一组。

代码

#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;   //求前k大的数的和
	cout << sum << endl;    
	
	sort(p + 1, p + 1 + k, cmp2);
	
	if(k == 1)   //如果只有一段,就直接输出n
	{
		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;   //对n进行特判
	
	return 0;
} 
上一篇 下一篇