题目链接: Max Sum Plus Plus.
大致题意
给定m和n,将n个数分成m段(可不取完),求出m段和的最大值。
1 ≤ n ≤ 1,000,000
思路
1.二维DP?
定义dp[i][j] : 选取第 j 个数字的情况下将前 j 个数字分为 i 段的最大和。
可分为两种情况:
1.第 j 个数不独立成段 : d[i][j-1]
2.第 j 个数独立成段:d[i-1][k] ,其中 k=i-1,i,…,j-1
故可导出递推式
d[i][j] = max(d[i][j-1],d[i-1][k]) + a[j]
然鹅由于m的范围没给,故直接二维dp会导致空间时间双爆炸
2.滚动数组优化
发现第 i 项只与 i-1与 i 有关,因此只需记录两个状态即可
d[t][j] = max(d[t][j-1],d[1-t][k]) + a[j] ,其中 k=i-1,i,…,j-1 ,t=1
然后就发现其实时间也可以优化(因为我们只需求最大和,不需要最大和发生的位置)
(见题面:But I`m lazy, I don’t want to write a special-judge module, so you don’t have to output m pairs of i and j, just output the maximal summation of sum(ix, jx)(1 ≤ x ≤ m) instead. ^ _ ^)
故我们可以用pre[j-1]记录 j-1 时的最大和。
得出递推式
d[t][j] = max(d[t][j-1],pre[j-1]) + a[j]
发现用一维就能够表示,得到最终的递推式(好耶):
d[j] = max(d[j-1],pre[j-1]) + a[j]
AC代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#define N 1000005
#define INF -0x3f3f3f3f
using namespace std;
int a[N];
int pre[N];
int dp[N];
int main()
{
int m,n,ans;
while(~scanf("%d%d",&m,&n))
{
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));
memset(pre,0,sizeof(pre));
for(int i=1; i<=m; i++)
{
ans = INF;
for(int j=i; j<=n; j++)
{
dp[j] = max(dp[j-1],pre[j-1]) + a[j];
pre[j-1] = ans;
ans = max(ans,dp[j]);
}
}
printf("%d\n",ans);
}
return 0;
}