1052 最大M子段和

1052 最大M子段和(51nod)

题目描述：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] 代表在j个数中选出i个子段的最大值 第j个数一定选
sum[i][j] 代表在j个数中选出i个子段的最大值 第j个数不一定选
dp[i][j] = max(dp[i][j-1]+val[j],sum[i-1][j-1]+val[j]);
第j个数要选的话 可以和val[j-1]在一个子段中 或者单独作为一个子段
但是根据dp数组的含义 我们知道单独作为一个子段 dp数组是没有办法更新的
所以我们引进一个新的数组 sum[i][j] 这时候 第j个数可以不选 也可以单独作为一个子段
sum数组的更新与dp有关
所以 sum[i][j] = max(dp[i][j],sum[i][j-1]);
我顺便用了一波滚动数组，感兴趣的小伙伴可以自己去看看。

附上代码：

#include<bits/stdc++.h>
#define cn getchar 
typedef long long LL;
const int MAXN=5010;

int n,m; 
int val[MAXN];
LL dp[2][MAXN],sum[2][MAXN];
void in(int &x){
	int f1=1;char ch=cn();x=0;
	while(ch<'0'||ch>'9'){if(ch=='-')f1=-1;ch=cn();}
	while(ch>='0'&&ch<='9'){x=x*10+(ch-'0');ch=cn();}
	x*=f1;
}
inline LL max(LL a,LL b) {return a<b?b:a;} 
 int main() {
 	in(n);in(m);
    for(int i=1;i<=n;i++)in(val[i]);
    
	for(int i=0;i<2;++i)
       for(int j=0;j<=n;++j)
         dp[i][j]=sum[i][j]=-1e9;
     
	for(int i=0;i<=n;++i) dp[0][i]=sum[0][i]=0;
    
	for(int i=1;i<=m;++i){
    	for(int j=1;j<=n;++j) {
           dp[i&1][j]=max(sum[(i-1)&1][j-1]+val[j],dp[i&1][j-1]+val[j]);
           sum[i&1][j]=max(dp[i&1][j],sum[i&1][j-1]);
   	    }
   	}
     printf("%lld\n",sum[m&1][n]);
     return 0;
 }