问题描述:
给定一串n个整数的序列,从中找出最大连续子序列的和,结果最小不能低于0,若小于0则直接返回0。
求解:
暴力法比较简单,就是从头到位把所有序列枚举一遍,然后找出其中最大的输出就行。
i 指向开头,j 指向结尾,k 从 i 到 j 遍历一遍。
核心代码:
int find_max_sequence(int* a, int n) {
int max = 0;
int thissum = 0;
for (int i = 0; i < n; i++) {
for (int j = i; j < n; j++) {
thissum = 0;
for (int k = i; k <= j; k++)
thissum += a[k];
if (thissum > max)
max = thissum;
}
}
return max;
}
但是这样似乎太简单了,而且时间复杂度n^3也高的离谱,我们需要对算法进行优化。
优化步骤一:
拥有最大和的连续子序列一定是正整数开头,即i指向的元素一定不能是负数。
优化一代码:
int find_max_sequence(int* a, int n) {
int max = 0;
int thissum = 0;
for (int i = 0; i < n; i++) {
for (int j = i; j < n; j++) {
thissum = 0;
if (a[i] < 0)
continue;
for (int k = i; k <= j; k++)
thissum += a[k];
if (thissum > max)
max = thissum;
}
}
return max;
}
优化步骤二:
这一次的优化我直接举一个例子。
再求-1到-7这个序列得出结果22,在求-1到9这个序列时只需要蓝色序列加上9就能得出结果,这样相当于去掉了一层循环,时间复杂度降到n^2。
优化步骤二代码:
int find_max_sequence(int* a, int n) {
int max = 0;
int thissum = 0;
for (int i = 0; i < n; i++) {
thissum = 0;
for (int j = i; j < n; j++) {
thissum += a[j];
if (a[i] < 0)
continue;
if (thissum > max)
max = thissum;
}
}
return max;
}
优化步骤三:
根据题目的要求,最大的连续子序列和一定大于等于零,那么如果某个子序列和降到0以下,就直接可以让thissum=0,即舍弃前面这一段小于零的序列,从下一个继续开始,这样又可以去掉一层循环,时间复杂度降为n。
优化步骤三代码:
int find_max_sequence(int* a, int n) {
int max = 0;
int thissum = 0;
for (int i = 0; i < n; i++) {
thissum += a[i];
if (thissum < 0)
thissum = 0;
if (max < thissum)
max = thissum;
}
return max;
}
优化之后的完整代码:
#include<stdio.h>
#include<iostream>
using namespace std;
int find_max_sequence(int* a, int n) {
int max = 0;
int thissum = 0;
for (int i = 0; i < n; i++) {
thissum += a[i];
if (thissum < 0)
thissum = 0;
if (max < thissum)
max = thissum;
}
return max;
}
int main() {
int a[] = { -6,2,4,-7,5,3,2,-1,6,-9,10,-2 };
int n = sizeof(a) / sizeof(int);
cout << find_max_sequence(a, n);
return 0;
}
通过以上优化我们可以看出,其实暴力法经过一些优化,其算法效率完全不输于贪心动归分治这些算法。