Week12作业 C -必做题 - 3 [HDU - 1024]

题目链接

https://vjudge.net/problem/HDU-1024

题目描述

n个数的序列a，从中选取m个互不相交的闭区间，使这些区间中的数的和最大，输出最大和(1 ≤ i ≤ n ≤ 1e6 , -32768 ≤ ai ≤ 32767)。

输入

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

输出

输出最大和。

基本思路

这是一个动态规划问题，关键是要定状态，找转移方程，如下表所示：

状态	定义分$f[i][j]$序列中前j个数选i个区间并以a[j]结尾的最大和
初始化	$$\begin{gathered} f[1][1]=a[1] \\ f[1][j]=\max (f[1][j-1]+a[j],a[j])(2\le j\le n) \\ f[j][j]={\sum }_{i=1}^j{a}_j=f[j-1][j-1]+a[j](2\le j\le n) \end{gathered}$$
转移过程	$f[i][j]=\max (f[i][j-1]+a[j],{\max }_{k=i-1}^{j-1}(f[i-1][k])+a[j])$
输出答案	${\max }_{k=m}^n(f[m][k])$
时间复杂度	$O(mn)$

由于状态转移时f[i][j]只与f[i][j-1]和f[i-1][k]有关，所以当计算分成i个区间的状态前可以将i-1个区间的状态数组f[i-1]存起来，而之前的i-2、i-3···可以去掉，于是可以去掉f数组的一个维度，从而将空间复杂度降到$O(n)$。

另外，由于本题输入数据量很大，若使用cin读入，应当关掉同步以提高效率。

完整代码

#include<iostream>
#include<algorithm>
using namespace std;
//f[i][j] = (f[i][j-1], max(f[i-1][k])) + a[j] 
long long a[1000005];
long long f[1000005];
long long lf[1000005];
long long maxn;
int n, m; 
int main ()
{
	ios::sync_with_stdio(false);
	while(cin>>m>>n)
	{
		for(int i=0;i<=n;i++)	f[i] = lf[i] = 0;
		for(int i=1;i<=n;i++)	cin>>a[i];
		for(int i=1;i<=n;i++)	f[i] = max(f[i-1], 0) + a[i]; 
		for(int i=2;i<=m;i++)
		{
			for(int j=i-1;j<=n;j++)	lf[j] = f[j];
			f[i] = f[i-1] + a[i];
			maxn = max(lf[i], lf[i-1]);
			for(int j=i+1;j<=n;j++)
			{
				f[j] = max(f[j-1], maxn) + a[j];
				maxn = max(maxn, lf[j]);
			} 
		}
		maxn = f[m];
		for(int i=m+1;i<=n;i++)	maxn = max(maxn, f[i]);
		cout<<maxn<<endl;	
	} 
	return 0;
}
/*
3 10 -5 2 2 -6 3 5 -1 6 1 -13
2 6 -1 4 -2 3 -2 3
2 7 8 -18 20 -7 8 -17 -16
2 9 9 -18 20 -7 -1 9 8 -17 -16
*/