CF1006B 题解

题意

给定长度为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;
} 
上一篇:Java学习——数据结构之约瑟夫问题


下一篇:Python之使用unittest模块测试