题目:输入一个整型数组,数组里有正数也有负数。数组中一个或连续的多个整数组成一个子数组。求所有子数组的和的最大值。要求时间复杂度为O(n)。
#include <functional>
using namespace std;
//定义一个全局变量标记输入是否无效
bool g_InvalidInput = false;
int FindGreatestSumOfSubArray(int* pData, int length)
{
//当输入无限时把标记置为true且返回0
if (pData == nullptr || length <= 0)
{
g_InvalidInput = true;
return 0;
}
g_InvalidInput = false;
//nCurSum储存当前和,nGreatestSum储存当前最大的和;
//当nCurSum<=0时我们舍弃它,并把当前的pData[i]赋给nCurSum
//否则nCurSum继续加上pData[i],并和nGreatestSum比较,把较大的赋给nGreatestSum
int nCurSum = 0;
int nGreatestSum = 0;
for (int i = 0;i < length;++i)
{
if (nCurSum <= 0)
nCurSum = pData[i];
else
nCurSum += pData[i];
if (nCurSum > nGreatestSum)
nGreatestSum = nCurSum;
}
return nGreatestSum;
}
// ====================测试代码====================
void Test(const char* testName, int* pData, int nLength, int expected, bool expectedFlag)
{
if (testName != nullptr)
printf("%s begins: \n", testName);
int result = FindGreatestSumOfSubArray(pData, nLength);
if (result == expected && expectedFlag == g_InvalidInput)
printf("Passed.\n");
else
printf("Failed.\n");
}
// 1, -2, 3, 10, -4, 7, 2, -5
void Test1()
{
int data[] = { 1, -2, 3, 10, -4, 7, 2, -5 };
Test("Test1", data, sizeof(data) / sizeof(int), 18, false);
}
// 所有数字都是负数
// -2, -8, -1, -5, -9
void Test2()
{
int data[] = { -2, -8, -1, -5, -9 };
Test("Test2", data, sizeof(data) / sizeof(int), -1, false);
}
// 所有数字都是正数
// 2, 8, 1, 5, 9
void Test3()
{
int data[] = { 2, 8, 1, 5, 9 };
Test("Test3", data, sizeof(data) / sizeof(int), 25, false);
}
// 无效输入
void Test4()
{
Test("Test4", nullptr, 0, 0, true);
}
int main(int argc, char* argv[])
{
Test1();
Test2();
Test3();
Test4();
return 0;
}