题目描述
又到了吃午饭的时间,你和你的同伴刚刚研发出了最新的GSS-483型自动打饭机器人,现在你们正在对机器人进行功能测试。
为了简化问题,我们假设午饭一共有N个菜,对于第i个菜,你和你的同伴对其定义了一个好吃程度(或难吃程度,如果是负数的话……)A[i],
由于一些技(经)术(费)限制,机器人一次只能接受一个指令:两个数L, R——表示机器人将会去打第L~R一共R-L+1个菜。
本着不浪费的原则,你们决定机器人打上来的菜,含着泪也要都吃完,于是你们希望机器人打的菜的好吃程度之和最大
然而,你善变的同伴希望对机器人进行多次测试(实际上可能是为了多吃到好吃的菜),他想知道机器人打M次菜能达到的最大的好吃程度之和
当然,打过一次的菜是不能再打的,而且你也可以对机器人输入-1, -1,表示一个菜也不打
输入描述:
第一行:N, M
第二行:A[1], A[2], ..., A[N]
输出描述:
一个数字S,表示M次打菜的最大好吃程度之和
示例1
输入
7 2
1 2 3 -2 3 -10 3
输出
10
说明
[1 2 3 -2 3] -10 [3]
示例2
输入
7 4
1 2 3 -2 3 -10 3
输出
12
说明
[1 2 3] -2 [3] -10 [3]
第四次给机器人-1, -1的指令
备注:
N <= 10^5 = 100000
|A[i]| <= 10^4 = 10000
10%数据M = 1
50%数据M <= 2
80%数据M <= 100
100%数据M <= 10^4 = 10000
思路:常规思路:dp[i][j]表示前i个数切分不超过j段的最大和,则dp[i][j]=min(dp[i-1][j],dp[k][j-1])+a[i],其中j-1<=k<i,这是很好理解的,dp[i-1][j]表示最后一段以j结尾的最大和,dp[k][j-1]表示第j个数开始新开辟了一段的最大和。
根据上述转移方程我们不难在O(m*n*n)的时间复杂度内解决,然而,题目的数据量使得我们无法通过这种方法解决这道题,另辟奇径:滚动数组!
我们发现在转移方程中第二维永远只和当前段数减一有关,因此我们其实是没必要保存这一维的,此外,我们仔细品一下这个转移方程,你们觉得dp[k][j-1]需要求吗?
其实我们在求dp[i][j]的时候已经求出来所有的dp[k][j-1]的值了,呢直接通过一个中间变量把这个值利用起来就好啦,这样我们能够在降低一维空间复杂度的基础上再降低一维时间复杂度!
然而上述优化完成后,最后还是只能过80%的数据,我们只能考虑骗数据的优化:将连续的正数或负数合并成一个数,从而缩小原数组的长度,最终AC!
#include<stdio.h>
#include<algorithm>
using namespace std;
#define maxn 100005
int n,m,a[maxn],dp[maxn],cnt,ans;
int main(void){
scanf("%d%d",&n,&m);
for(int i=0,x;i<n;i++){
scanf("%d",&x);
if(cnt==0)
a[++cnt]=x;
else{
if(a[cnt]*x>0)
a[cnt]+=x;
else
a[++cnt]=x;
}
}
n=cnt;
for(int i=1;i<=m;i++){
int sum=0;
for(int j=1;j<i;j++)
sum+=a[j];
ans=sum;
for(int j=i;j<=n;j++){
sum=max(sum, dp[j-1])+a[j];
dp[j-1]=ans;
ans=max(ans, sum);
}
ans=max(ans, sum);
}
printf("%d\n",ans);
return 0;
}