LintCode 402: Continuous Subarray Sum

LintCode 402: Continuous Subarray Sum

题目描述

给定一个整数数组，请找出一个连续子数组，使得该子数组的和最大。输出答案时，请分别返回第一个数字和最后一个数字的下标。（如果两个相同的答案，请返回其中任意一个）

样例

给定[-3, 1, 3, -3, 4], 返回[1,4].

Thu Feb 23 2017

思路

本题有很多解法，最巧妙的解法就是一遍扫描记忆的方法了，时间复杂度为\(O(n)\)。

用一个变量记录连续累加的和，当和为负数时，变量清零，从下一个数字开始累加记录。

在这里只需要注意一下一些小细节，比如如何记录最大子数组的起始位置，以及处理一下数组中所有的数都是负数的情况。

代码

// 连续子数组求和
vector<int> continuousSubarraySum(vector<int>& A)
{        
    int max_sum = -1, sum = 0, start;
    int is_all_negative = 1, max_num = -9e5, max_num_ind = -1
    vector<int> ans(2);
    for (int i = 0; i < A.size(); ++i)
    {
        if (sum <= 0) start = i;
        sum += A[i];
        if (sum < 0) sum = 0;
        else if (sum > max_sum)
        {
            is_all_negative = 0;
            max_sum = sum;
            ans[0] = start;
            ans[1] = i;
        }

        if (max_num < A[i])
        {
            max_num = A[i];
            max_num_ind = i;
        }
    }

    if (is_all_negative)
    {
        ans[0] = ans[1] = max_num_ind;
    }
    return ans;
}