C - 必做题 - 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)

题意分析

用dp[j]表示从前j个元素中找i段最大子串和，pre[j]记录每一轮j不断变化时得到的最大dp值。状态转移方程为dp[j]=max(dp[j-1],pre[j-1])+a[j],表示要么将a[j]作为单独的一个段，要么a[j]和后续元素一起组成一个段。

通过代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

#define maxn 1000010
const int inf = 0x3f3f3f3f;
int dp[maxn], pre[maxn], a[maxn];
int n, m;

int main()
{
	while (scanf("%d%d", &m, &n)!=EOF)
	{
		memset(dp, 0, sizeof(dp));
		memset(pre, 0, sizeof(pre));
		for (int i = 1; i <= n; ++i)
			scanf("%d", &a[i]);
		int maxx;
		for (int i = 1; i <= m; ++i)
		{
			maxx = -inf;
			for (int j = i; j <= n; ++j)
			{
				dp[j] = max(dp[j - 1], pre[j - 1]) + a[j];
				pre[j - 1] = maxx;//为i+1做准备
				if (dp[j] > maxx)
					maxx = dp[j];//保存前j个段数为i时的最大值
			}
		}
		printf("%d\n", maxx);
	}
	return 0;
}