HDU 1024：Max Sum Plus Plus（DP）

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 39886 Accepted Submission(s): 14338

Problem Description

Now I think you have got an AC in Ignatius.L’s “Max Sum” problem. To be a brave ACMer, we always challenge ourselves to more difficult problems. Now you are faced with a more difficult problem.

Given a consecutive number sequence $S_1,S_2,S_3,S_4...S_x,...S_n(1\le x\le n\le 1,000,000,-32768\le S_x\le 32767)$. We define a function $sum(i,j)=S_i+...+S_j(1\le i\le j\le n)$.

Now given an integer $m(m>0)$, your task is to find m pairs of i and j which make $sum(i_1,j_1)+sum(i_2,j_2)+sum(i_3,j_3)+...+sum(i_m,j_m)$ maximal ($i_x\le i_y\le j_x$ or $i_x\le j_y\le j_x$ is not allowed).

But I`m lazy, I don’t want to write a special-judge module, so you don’t have to output m pairs of $i$ and $j$, just output the maximal summation of $sum(i_x,j_x)(1\le x\le m)$ instead.

Input

Each test case will begin with two integers $m$ and $n$, followed by $n$ integers $S_1,S_2,S_3...S_n$.
Process to the end of file.

Output

Output the maximal summation described above in one line.

Sample Input

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

Sample Output

6
8

题意

将一个长度为$n$的数组分成不相交的$m$段，求这$m$段的和的最大值

思路

状态：$dp[i][j]$表示在前$j$个数中取出$i$段的最大和

状态转移方程：$dp[i][j]=max(dp[i-1][k],dp[i][j-1])+num[j](i-1\le k\le j-1)$

由于$m$范围未知，$n\le 10^6$，所以二维的dp方程无论是在时间上还是在空间上都是不允许的。

那么我们就需要对这个方程进行优化：

不难发现当前状态只与两个状态有关：

1. 第$j$个数和前$j-1$个数在一段里
2. 第$j$个数和前$j-1$个数不在一段里。

根据这一点，我们把状态降成一维的数组，$dp[j]$表示前$j$个数分$i$段时的最大和，然后用$sum[j-1]$来表示状态一的前$j-1$个数在前$i-1$段的最大和，$dp[j-1]$表示状态二的前$j-1$个数在前$i$段的最大和。

当前状态的转移方程为：$dp[j]=max(dp[j-1],sum[j-1])+num[j]$，持续更新dp与sum数组的值

AC代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <limits.h>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <string>
#include <time.h>
#define ll long long
#define ull unsigned long long
#define ms(a,b) memset(a,b,sizeof(a))
#define pi acos(-1.0)
#define INF 0x7f7f7f7f
#define lson o<<1
#define rson o<<1|1
#define bug cout<<"-------------"<<endl
#define debug(...) cerr<<"["<<#__VA_ARGS__":"<<(__VA_ARGS__)<<"]"<<"\n"
const double E=exp(1);
const int maxn=1e6+10;
const int mod=1e9+7;
using namespace std;
int a[maxn];
int dp[maxn];
int sum[maxn];
int main(int argc, char const *argv[])
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	#ifndef ONLINE_JUDGE
	    freopen("in.txt", "r", stdin);
	    freopen("out.txt", "w", stdout);
	    double _begin_time = clock();
	#endif
	int k,n;
	while(cin>>k>>n)
	{
		int res;
		for(int i=1;i<=n;i++)
			cin>>a[i];
		ms(dp,0);
		ms(sum,0);
		for(int i=1;i<=k;i++)
		{
			res=-INF;
			for(int j=i;j<=n;j++)
			{
				dp[j]=max(sum[j-1],dp[j-1])+a[j];
				sum[j-1]=res;
				res=max(res,dp[j]);
			}
		}
		cout<<res<<endl;
	}
	#ifndef ONLINE_JUDGE
	    long _end_time = clock();
	    printf("time = %lf ms.", _end_time - _begin_time);
	#endif
	return 0;
}