解题思路:
题目中也没给m的范围,还以为可以O(n),结果只能O(nm)做。
dp[i][j]表示前j个数字分成i段,且a[j]在最后一段的最大和,那么转移方程为:
dp[i][j]=max(dp[i][j-1]+a[j],max(dp[i-1][k])+a[j]),i
≤
j
≤
n,i-1
≤
k
<
<script type="math/tex" id="MathJax-Element-12"><</script>j。
但这样时间空间都会超。
注意到dp[i]这一层只能由dp[i-1]一层转移,且max(dp[i-1][k])可以顺着j的循环维护,所以可以对于每一层i的循环,可以用mx[j]表示对于j合法的dp[i-1][k]的最大值,那么转移方程就变成了:
dp[j]=max(dp[j-1]+a[j],mx[j-1]+a[j])。
循环m次,最后取最大值即可。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<queue>
#define ll long long
using namespace std;
int getint()
{
int i=0,f=1;char c;
for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
if(c=='-')f=-1,c=getchar();
for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
return i*f;
}
const int N=1000005;
const ll INF=1e17;
int n,m,a[N];
ll dp[N],mx[N];
int main()
{
//freopen("lx.in","r",stdin);
while(scanf("%d%d",&m,&n)!=EOF)
{
memset(mx,0,sizeof(mx));
for(int i=1;i<=n;i++)
a[i]=getint();
ll tmp;
for(int i=1;i<=m;i++)
{
tmp=-INF;
for(int j=i;j<=n;j++)
{
dp[j]=max(dp[j-1]+a[j],mx[j-1]+a[j]);
mx[j-1]=tmp;
tmp=max(tmp,dp[j]);
}
}
cout<<tmp<<'\n';
}
return 0;
}