31. 连续子数组的最大和

题目描述

输入一个整型数组，数组里有正数也有负数。数组中一个或连续的多个整数组成一个子数组。求所有子数组的和的最大值。要求时间负责度为O(n)。

解析

1. 解法一：
看到这个题目，我们首先想到的是求出这个整型数组所有连续子数组的和，长度为n的数组一共有 n(n+2)/2个子数组，因此要求出这些连续子数组的和最快也需要O(n^2)的时间复杂度。但是题目要求的O(n)的时间复杂度，因此上述思路不能解决问题。
2. 解法二：
看到O(n)时间复杂度，我们就应该能够想到我们只能对整个数组进行一次扫描，在扫描过程中求出最大连续子序列和以及子序列的起点和终点位置。假如输入数组为{1,-2,3,10,-4,7,2,-5}，我们尝试从头到尾累加其中的正数，初始化和为0，第一步加上1，此时和为1，第二步加上-2，此时和为-1，第三步加上3，此时我们发现-1+3=2，最大和2反而比3一个单独的整数小，这是因为3加上了一个负数，发现这个规律以后我们就重新作出累加条件：如果当前和为负数，那么就放弃前面的累加和，从数组中的下一个数再开始计数。
3. 解法三：应用动态规划法

主要代码实现

int FindgGreatestOfSubArray(int* pData, int nLength){
    //传参有效性检测
    if(pData == NULL || nLength <= 0){
        return 0;
    }
    int nCurSum = 0;
    int nGreatestSum = 0;

    for(int i=0; i<nLength; i++){
        if(nCurSum < 0)
            nCurSum = pData[i];     //抛弃i下标之前的数组元素
        else
            nCurSum += pData[i];
        if(nCurSum > nGreatestSum)
            nGreatestSum = nCurSum;

    }
    return nGreatestSum;
}

测试用例

// ====================测试代码====================
void Test(char* testName, int* pData, int nLength, int expected, bool expectedFlag)
{
    if(testName != NULL)
        printf("%s begins: \n", testName);

    int result = FindgGreatestOfSubArray(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", NULL, 0, 0, true);
}

int main(void)
{
    Test1();
    Test2();
    Test3();
    Test4();
    system("pause");
    return 0;
}

改进

保存最大和对应的子数组：修改上面的主要代码，当重置当前数组和的时候，记录起始数组下标，修改数组最大值的时候，记录当前最大值的数组下标。（在函数中采用引用传递参数，第一次使用引用变量）

int FindgGreatestOfSubArray(int* pData, int nLength, int& a, int& b){
    //传参有效性检测
    if(pData == NULL || nLength <= 0){
        g_InvalidInput = true;
        return 0;
    }
    int nCurSum = 0;
    int nGreatestSum = 0;

    for(int i=0; i<nLength; i++){
        if(nCurSum < 0){
            nCurSum = pData[i];     //抛弃i下标之前的数组元素
            a = i;
        }
        else
            nCurSum += pData[i];
        if(nCurSum > nGreatestSum){
            nGreatestSum = nCurSum;
            b = i;
        }
    }
    return nGreatestSum;
}

参考博文

面试题31：连续子数组的最大和