Codeforce - 1006B - Polycarp's Practice （贪心）

题目链接：http://codeforces.com/problemset/problem/1006/B

题意

将一个长度为n的数组划分为k段。求每段中的最大值相加后的和的最大值 与 划分后的k段每一段有几个数据。

                                                                                                                                                                                   

题解

为了保证每段最大值相加后和的最大值最大，则应将所有数据排列好后，前k个数据就为所要求的每段中的最大值。接下来就是求区间长度。可将前k个数据的下标保存下来，用于求区间长度。第一个区间为以其下标往左延申到尽头，往右延申到下一个下标（应考虑无下一个坐标的情况）。剩余区间均为对应下标向右延申到下一个坐标（最后一个区间延申到 n ）

• AC Code

  #include <iostream>     
  #include <cstdio>
  #include <cstring>
  #include <cmath>
  #include <algorithm>
  using namespace std;
  
  struct Data
  {
  	int x;
  	int id;
  };
  bool cmp1(Data a, Data b)
  {
  	if(a.x > b.x) return true;
  	else if(a.x == b.x)
  	{
  		return a.id > b.id;
  	}
  	else return false;
  }
  bool cmp2(int x, int y)
  {
  	return x < y;
  }
  Data st[2000+5];
  int a[2000+5];
  int b[2000+5];
  int main()
  {
  	int n, k;
  	cin>> n >> k;
  	for(int i=1; i<=n; i++)
  	{
  		cin>> st[i].x;
  		st[i].id = i;
  	}
  	sort(st+1, st+1+n, cmp1);
  	int sum = 0;
  	for(int i=1; i<=k; i++)
  	{
  		sum = sum + st[i].x;
  		a[i] = st[i].id;
  	}
  	sort(a+1, a+1+k, cmp2);
  	for(int i=1; i<=k; i++)   //根据下标进行求划分区间 （关键在于第一个区间和最后一个区间） 
  	{
  		if(i==1)
  		{
  			b[i] = a[i];
  			if(a[2]!=0) b[i] = b[i] + a[2]-a[1]-1; //重点 
  			if(a[2]==0) b[i] = n; 
  		}
  		else if(1<i && i<k)
  		{
  			b[i] = a[i+1] - a[i];
  		}  
  		else b[i] = n - a[i] + 1;
  	}
  	cout<< sum << endl;
  	for(int i=1; i<=k; i++)
  	{
  		if(i==1) printf("%d",b[i]);
  		else     printf(" %d",b[i]);
  	}
  	printf("\n");
  	
  	
  	return 0;
  }