题目描述:
N个整数组成的序列a[1],a[2],a[3],…,a[n],将这N个数划分为互不相交的M个子段,并且这M个子段的和是最大的。如果M >= N个数中正数的个数,那么输出所有正数的和。
例如:-2 11 -4 13 -5 6 -2,分为2段,11 -4 13一段,6一段,和为26。
输入
第1行:2个数N和M,中间用空格分隔。N为整数的个数,M为划分为多少段。(2 <= N , M <= 5000)
第2 - N+1行:N个整数 (-10^9 <= a[i] <= 10^9)
输出
输出这个最大和
输入样例
7 2
-2
11
-4
13
-5
6
-2
输出样例
26
分析:设dp[i][j]表示用了前i个数,组成的字段数为j的最大和。公式很容易推导出,dp[i][j]=max(dp[i][j],dp[i-1][j]+a[i],dp[t][j-1]+ai).因为会TLE,所以我们用mx[i][j]表示第i行,前j个数的最大值。方程就可以优化到n^2为dp[i][j]=max(dp[i][j],dp[i-1][j]+a[i],mx[i-1][j-1]+a[i]);又因为会爆内存,又因为当前层的值只会与上一层有关,所以在我们要用一个滚动数组来优化
那么具体代码如下:
#include<bits/stdc++.h>
using namespace std;
#define LL long long
LL dp[2][5010];
LL n,m;
LL mx[2][5010];
LL a[100001];
int main(){
scanf("%lld %lld",&n,&m);
for (LL i=1;i<=n;i++) scanf("%lld",&a[i]);
LL lm=0;
LL ans=0;
for (LL i=1;i<=n;i++)
for (LL j=1;j<=m;j++){
LL summ=dp[(i-1)&1][j]+a[i];
LL sumn=mx[i-1&1][j-1]+a[i];
dp[i&1][j]=max(summ,sumn);
mx[i&1][j]=max(mx[(i-1)&1][j],dp[i&1][j]);
if (j==m) ans=max(ans,dp[i&1][j]);
}
printf("%lld",ans);
return 0;
}