Polycarp's Practice

Polycarp’s Practice

题目大意：

给定一个长度为n的数列，将它分成k段（不改变原序）。
求一种分法，使得每段中的最大值之和最大。
（分法的描述形式为每段长度）
Special Judge

输入格式：

n k
a1 a2 a3 … an

输出格式：

maxsum
t1 t2 t3 … tk

数据范围

1<=k<=n<=2000 (int)
1<=ai<=2000 (int)

样例输入输出：

Input1:

8 3
5 4 2 6 5 1 9 2

Output1:

20
3 2 3

Input2:

5 1
1 1 1 1 1

Output2:

1
5

Input3:

4 2
1 2000 2000 2

Output3:

4000
2 2

样例说明：

样例1：分3段（有多种分法）,例如[5,4,2],[6,5,1],[9,2]。
样例2：分1段=>直接最大值1
样例3：分2段，每段找一个2000

简单题、解：

从maxsum入手，由题+分析样例，maxsum=前k大值之和，只需要让每个区间都包含一个前k大值即可。

1. 前k大值==>sort
2. sort后的元素位置问题==>struct+cmp
3. 自行构造包含前k大值的区间

参考程序：

#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 2000

struct node {
    int v,p;
}a[MAXN+5];
int n,k,ans;
int b[MAXN+5];

bool cmp(node x,node y) {
    return x.v>y.v;
}
int main() {
    scanf("%d %d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i].v),a[i].p=i;
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=k;i++)
        ans+=a[i].v,b[i]=a[i].p;//用b存位置
    printf("%d\n",ans);
    sort(b+1,b+k+1);//b[i]：第i个区间的起点
    b[1]=1,b[k+1]=n+1;//限定范围保险
    for(int i=1;i<k;i++)
        printf("%d ",b[i+1]-b[i]);
    printf("%d\n",b[k+1]-b[k]);
    return 0；
}