牛客假日团队赛1——J——分组（DP）

链接：https://ac.nowcoder.com/acm/contest/918/J
来源：牛客网
 

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld

题目描述

在Farmer John最喜欢的节日里，他想要给他的朋友们赠送一些礼物。由于他并不擅长包装礼物，他想要获得他的奶牛们的帮助。你可能能够想到，奶牛们本身也不是很擅长包装礼物，而Farmer John即将得到这一教训。

Farmer John的N头奶牛（1≤N≤10^4）排成一行，方便起见依次编号为1…N。奶牛i的包装礼物的技能水平为si。她们的技能水平可能参差不齐，所以FJ决定把她的奶牛们分成小组。每一组可以包含任意不超过K头的连续的奶牛（1≤K≤10^3），并且一头奶牛不能属于多于一个小组。由于奶牛们会互相学习，这一组中每一头奶牛的技能水平会变成这一组中水平最高的奶牛的技能水平。

请帮助FJ求出，在他合理地安排分组的情况下，可以达到的技能水平之和的最大值。

输入描述:

输入的第一行包含N和K。以下N行按照N头奶牛的排列顺序依次给出她们的技能水平。技能水平是一个不超过10^5的正整数。

输出描述:

输出FJ通过将连续的奶牛进行分组可以达到的最大技能水平和。

示例1

输入

复制

7 3
1
15
7
9
2
5
10

输出

复制

84

说明

在这个例子中，最优的方案是将前三头奶牛和后三头奶牛分别分为一组，中间的奶牛单独成为一组（注意一组的奶牛数量可以小于K）。这样能够有效地将7头奶牛的技能水平提高至15、15、15、9、10、10、10，和为84。

动态规划，规划的内容就是如何分段，于是可以想到类似最优子结构的一个规划

令dp[i]表示前i个的最大值，j表示枚举最后一个分组的起点，则

dp[i]=max(dp[i],dp[i-(i-j+1)]+mmax*(i-j+1))

理解起来就是，当前i节点的最大值，一定是往前推j个数，将这一段变为最大数值值，然后再加上前i-(i-j+1)的最优情况。所以枚举j即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>
using namespace std;

const int maxn=1e4+10;
int a[maxn];
int dp[maxn];

int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    memset(dp,0,sizeof(dp));
    for(int i=1;i<=n;i++){
        int mmax=-1;
        for(int j=i;j>i-k&&j>=1;j--){
            mmax=max(mmax,a[j]);
            dp[i]=max(dp[i],dp[i-(i-j+1)]+mmax*(i-j+1));
        }
    }
    printf("%d\n",dp[n]);
    return 0;
}