题目形式:给定一个数组,求其最大连续子数组和。(扩展题型还要求指出起始位置)
例如:arr = [1, 5, -10, 2, 5, -3, 2, 6, -3, 1]。输出为:12。对应的连续子数组为[2, 5, -3, 2, 6]。
代码:
注意点:
1、考虑全为负数情况。
2、有很多博文代码跑出来结果健壮性不够,此处有点到问题所在和临时解决办法。
3、建议先从只求最大连续和版本看,此版本简洁明了。
4、其它情况的测验(全为非正整数及特殊情况),打开和屏蔽相关注释即可。
/******************************************************************************
*
* 求解子数组最大连续和 V1.0
*
* ===========================================================================
*
* 文 件 名:test.cpp
* 模块名称:求解子数组最大连续和
* 功能说明:求解子数组最大连续和
* 作 者:LiuHuan
* 创建日期:2019-09-02
* 版 本 号:1.0
* 修改历史:
*
* 修改日期 姓名 内容
* ---------------------------------------------------------------------------
* 2019-09-02 LiuHuan 初始版本
*
******************************************************************************/
#include <iostream>
#include <vector>
#include <algorithm>
#include <time.h>
using namespace std;
// 返回正负
static inline int Symbol(int num)
{
return (num == 0 ? -1 : 1);
}
class Solution
{
public:
// 只求最大连续和版本, 时间复杂度O(n)
int MaxSum(const vector<int>& vec)
{
if (vec.size() == 0) return 0;
int maxCurrent = 0;
int maxAll = vec[0]; // 想下为何不能为0而是vec[0]? (因为序列有可能全为负数!)
for (auto item : vec)
{
maxCurrent = item > (maxCurrent + item) ? item : (maxCurrent + item);
maxAll = maxAll > maxCurrent ? maxAll : maxCurrent;
}
return maxAll;
}
// 重载带起始位置版本
int MaxSum(const vector<int>& vec, int& startPos, int& endPos)
{
if (vec.size() == 0) return 0;
int maxCurrent = 0;
int maxAll = vec[0];
bool isAllNegative = true; // 判断是否全部为负数
for (size_t i = 0; i < vec.size(); i++)
{
isAllNegative = vec[i] > 0 ? false : isAllNegative; // 如果考虑全为负数情况, 需要加此逻辑
maxCurrent += vec[i];
if (maxCurrent > maxAll)
{
maxAll = maxCurrent;
endPos = i; // 暂标endPos为当前位置
}
if (maxCurrent < 0) // 当前和小于零了需要重置startPos, endPos
{
maxCurrent = 0;
/* if (i <= (vec.size() - 2) && vec[i + 1] > 0) // 有的博客这样写, 注意是不健壮的(bug几率小, 后面有补救措施)
比如:9, -2, 6, 7, -7, 9, 3, -4, -5, 2, -6, -2, -7, -5, -3, -4, 0, -2, -2, 1 这组序列试试? 会出现StartPos = 19, endPos = 6
*/
if (i <= (vec.size() - 2) && vec[i + 1] > 0) // 注意不要越界
{
startPos = i + 1;
}
}
}
// bug补救措施, 虽然几率小, 目前只想到这种方式补救, 如有优化解, 烦请告知
if (startPos > endPos && !isAllNegative) // endPos与MaxAll恒定正确
{
int sumTmp = 0;
for (size_t i = endPos; i >= 0; i--)
{
sumTmp += vec[i];
if (sumTmp == maxAll) // 从endPos往前遍历, 满足后退出
{
startPos = i;
break;
}
}
}
if (isAllNegative) startPos = endPos; // 全为负数则最大连续和就是最大值
return maxAll;
}
};
int main()
{
srand((unsigned int)time(NULL));
vector<int> vec;
Solution s;
int startPos = 0;
int endPos = 0;
// 生成随机整数序列
for (size_t i = 0; i < 20; i++)
{
int value = Symbol(rand() % 2) * (rand() % 10); // 随机值在[10]以内心算快
//int value = -1 * (rand() % 10); // 随机序列全为非正数的情况
vec.push_back(value);
}
// vec = {9, -2, 6, 7, -7, 9, 3, -4, -5, 2, -6, -2, -7, -5, -3, -4, 0, -2, -2, 1}; // 用于测验63行的特殊情况
// 打印生成随机序列
cout << "vec:\t";
for_each(vec.cbegin(), vec.cend(), [](int num){ cout<< num << ", "; });
int res = s.MaxSum(vec, startPos, endPos);
// int res = s.MaxSum(vec);
cout << "\nMaxSum: " << res << endl;
// 使用Max(const vector<int>& vec)版本时, 下述输出不具参考性
cout << "startPos: " << startPos << ", endPos: " << endPos << endl;
cout << "Child Sequence: [ ";
for (size_t i = startPos; i < endPos; i++)
{
cout << vec[i] << ", ";
}
cout << vec[endPos] << " ]" << endl;
return 0;
}