week12 作业C必做题3

必做题 - 3

东东每个学期都会去寝室接受扫楼的任务，并清点每个寝室的人数。
每个寝室里面有ai个人(1<=i<=n)。从第i到第j个宿舍一共有sum(i,j)=a[i]+…+a[j]个人
这让宿管阿姨非常开心，并且让东东扫楼m次，每一次数第i到第j个宿舍sum(i,j)
问题是要找到sum(i1, j1) + … + sum(im,jm)的最大值。且ix <= iy <=jx和ix <= jy <=jx的情况是不被允许的。也就是说m段都不能相交。
注：1 ≤ i ≤ n ≤ 1e6 , -32768 ≤ ai ≤ 32767 人数可以为负数。。。。(1<=n<=1000000)

Input

输入m，输入n。后面跟着输入n个ai

Output

输出最大和

Sample Input

1 3 1 2 3
2 6 -1 4 -2 3 -2 3

Sample Output

6
8

我的思路：
这道题是要找到一个序列中多段连续且无重叠的子串的和的最大值。需要用到区间DP，这里我还进行空间复杂度的优化。列出状态转移方程，会发现每一行的数值只与其上一行有关，所以我只需要记录两行，形成滚动数组进行计算即可。对于dp方程，因为是要连续其分段加和最大，所以整体思路就是从小区间加到大区间，dp[k][j]就是前j个数取k段加和的最大值。

我的总结：
区间DP需要灵活运用，对于数据量大的题，要做一定的算法优化。

我的代码：

#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;


const int maxn = 0x3f3f3f3f;
int n,m;
long long a[1000005],dp[2][1000005],pre,ans;

int main()
{
	while(~scanf("%d%d",&m,&n))
	{
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	for(int j=0;j<2;j++) 
		for(int i=0;i<=n;i++)
			dp[j][i]=0;
	for(int i=1,k=1;i<=m;i++,k^=1)
	{
		dp[k][i-1] = -maxn;
		pre = -maxn;
		for(int j=i;j<=n-m+i;j++)
		{
			pre = max(pre,dp[k^1][j-1]);
			dp[k][j] = max(dp[k][j-1],pre)+a[j]; 
			//printf("%d:%d\n",j,dp[k][j]);
		}
	}
	ans = -maxn;
	for(int i=m;i<=n;i++)
	{
		ans = max(ans,dp[m&1][i]);
	}
	printf("%lld\n",ans);
	}
	return 0;
}