程序设计思维与实践 Week12 作业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)

Input

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

Output

输出最大和

Example

Input
1 3 1 2 3
2 6 -1 4 -2 3 -2 3
Output
6
8

思想

• 若用dp[i][j]表示前j个数取出i段得到的最大值，有两种情况：①第j个连在第j-1个所在的段的后面 ②第j个为新的一段的第一个数字。
• 状态转移方程为dp[i][j]=max(dp[i][j-1]+a[j],dp[i-1][k]+a[j]) ， i-1<=k<=j-1
• n^3会炸掉
• 所以用mx[j]表示前一个状态中1–j的dp的最大值，状态转移方程改为：dp[j] = max(dp[j-1] + a[j] ,mx[j-1]+a[j])

代码

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <queue>
using namespace std;
const int N=3000000; 
const int inf=0x3f3f3f3f;
int dp[N],a[N],mx[N]; 
int main()
{
	int m,n;
	int mxx;
	while(~scanf("%d%d",&m,&n))
	{
		memset(dp,0,sizeof(dp));
		memset(mx,0,sizeof(mx));
		for(int i=1;i<=n;i++)
			scanf("%d",&a[i]);
		for(int i=1;i<=m;i++)  //mx[i]是指第j个元素前划分成i-1份后的最大值
		{
			mxx=-inf;
			for(int j=i;j<=n;j++)  //要分成i组则至少需要i个数，所以j从i开始循环
			{
				dp[j]=max(dp[j-1]+a[j],mx[j-1]+a[j]);
				mx[j-1]=mxx;
				mxx=max(dp[j],mxx);
			}
		 } 
		printf("%d\n",mxx);
	}
	return 0; 
 }