C-必做题-3

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 处理到 EOF

Output

输出最大和 

Sample Input

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

Sample Output

6
8 

Hint

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

二、思路与算法

本题主要利用了DP，如果用二维数组的话，dp[i][j]表示前 i 段查询到第 j个数时的最大和，
dp[i][j]=max(dp[i][j−1]+a[j],max(dp[i−1][k])+a[j])，其中i<=k<j。
但是因为数据非常大，1e6 x 1e6无法承受，所以又加入了滚动数组。
因为在计算第 i 行的时候，之前的数据只需要知道max(dp[i-1][k])，所以用pm数组记录i、k不同时的max(dp[i-1][k])值。
pre_max算作一个中间变量。

三、代码实现

#include<cstdio>
#include<algorithm>
using namespace std;

long long m=0,n=0;
long long a[1000100]={0};
long long pre_max;
long long pm[1000100]={0};
long long dp[1000100]={0};   //DP数组 

void init(){
	for(int i=1;i<=n;i++){
		pm[i]=0;	dp[i]=0;	a[i]=0;
	}
}

int main(){
	while(scanf("%d %d",&m,&n)!=EOF){
		init();
		for(int i=1;i<=n;i++){
			scanf("%lld",&a[i]);
		}
		
		for(int i=1;i<=m;i++){
			pre_max=-(1e9+1e8);
			for(int j=i;j<=n;j++){
				dp[j]=a[j]+max(dp[j-1],pm[j-1]);
				pm[j-1]=pre_max;
				pre_max=max(pre_max,dp[j]);
			}
		}
		
		printf("%lld\n",pre_max);
	}
	return 0;
} 

四、经验与总结

1. dp这类型题目的结构还是比较相似的，一个dp数组，一个前驱数组，还有其他变量。动态规划最核心的总是前后关系的方程，基本推出方程之后，dp等，都是用来实现递推这个方程的，一步步推出最终要求的答案。