8.1 “最大连续合”问题_算法优化

写在前面

算法、源码来自《算法竞赛入门经典》第2版 P221，作者刘汝佳。

题目描述

最大连续和问题。 给出一个长度为n的序列A₁,A₂,…,A_n，求最大连续和。
换句话说，即要我们找到一段连续子序列，保证该子序列所有元素之和最大。

初始版本1.0

• 算法思想：枚举子序列的起点和终点，计算每种情况的和;
• 渐进时间复杂度：O（n³）；

#include <iostream>

using namespace std;

const int kLen = 6;

int main() {
	int A[kLen] = {1,2,-1,-4,3,6};
	int ans =  A[0];								//初始化第一个元素为解 
	for (int i = 0; i < kLen; i++) {
		int sum = 0;
		for (int j = i; j < kLen; j++) {
			sum +=  A[j];
			if (sum > ans)	ans = sum;
		}
	}
	cout << ans << endl;
	
	return 0;
}

优化版本2.0

• 算法思想：采用 “连续子序列的和等于两个前缀和之差” 的思想，提前递推求得序列的所有前缀和，后续只需两层循环即可求解；
• eg. 序列：{1，2，3，4，5}，子序列{4，5}之和 = 前缀和{1，2，3，4，5} - 前缀和{1，2，3}。
• 公式化：设S_i = A₁ + A₂ + … + A_i，则有：A_i + A_i+1 + … + A_j = S_j - S_i-1
• 用该思想优化后，渐进时间复杂度：O（n²）

#include <iostream>

using namespace std;

const int kLen = 6;

//eg. A[5] = {1,2,3,4,5};
//S[0] = 0;  S[1] = S[0] + 1;  S[2] = S[1] + 2 
 
int main() {
	int A[kLen] = {1,2,-1,-4,3,6};
	int ans =  A[0];					//初始化答案为第一个元素 
	int S[kLen + 1] = {0};				//S[0] = 0
	for (int i = 1; i <= kLen; i++) 
		S[i] = S[i - 1] + A[i - 1];		//递推求得每个位置的前缀和。
	
	for (int i = 1; i <= kLen; i++) {
		for (int j = i; j <= kLen; j++) {
			int sum = S[j] - S[i - 1];
			if (sum > ans) ans = sum; 
		}
	}
	cout << ans << endl;
	
	return 0;
}