Week12 作业C - 必做题 - 3

1.题意

东东每个学期都会去寝室接受扫楼的任务，并清点每个寝室的人数。
每个寝室里面有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)

2.样例

Input

输入m，输入n。后面跟着输入n个ai 处理到 EOF

Output

输出最大和

Sample Input

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

Sample Output

6
8

Hint

数据量很大，需要scanf读入和dp处理。

3.解题思路

1. 用数组f[i][j]表示前j个宿舍选了i段，则状态转移方程为f[i][j]=max{f[i][j-1]+a[j],f[i-1][k]+a[j]}，前者表示此宿舍与之前为一段，后者表示是新的一段，由于数据限制所以我们要进行降维处理，用另一个数组pmf记录前i-1段中的最大值
2. 要注意在更新pmf时ans的顺序

4.AC代码

#include<stdio.h>
#include<cstring>
#include<algorithm>
using namespace std;
const int M=1e6+5;
const int INF=1e8+5;
int a[M];
int f[M]={0};
int pmf[M]={0};
int main()
{
	int m,n,ans;
	while(~scanf("%d%d",&m,&n))
	{
		memset(f,0,sizeof(f));
		memset(pmf,0,sizeof(pmf));
		a[0]=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
		}
		for(int i=1;i<=m;i++)
		{
			ans=-INF;
			for(int j=i;j<=n;j++)
			{
				f[j]=max(f[j-1]+a[j],pmf[j-1]+a[j]);
				pmf[j-1]=ans;
				ans=max(ans,f[j]);
			}
		}
		printf("%d\n",ans);
	}
}