[LintCode] Continuous Subarray Sum 连续子数组之和

简介:

Given an integer array, find a continuous subarray where the sum of numbers is the biggest. Your code should return the index of the first number and the index of the last number. (If their are duplicate answer, return anyone)

Give [-3, 1, 3, -3, 4], return [1,4].

这道题跟LeetCode上的那道Maximum Subarray很类似,不同之处在于这道题让返回最大子数组的范围坐标,而之前那道题只需要返回最大和即可,所以这道题我们要及时更新start和end变量,分别为子数组的起始和结束位置。我们用curSum来记录当前位置的累计和,如果某一个位置之前的累计和为负数,那么我们直接从当前位置开始重新算即可,因为加上负数还不如不加,此时将start和end都更新为当前位置i;如果之前的累计和大于等于0,那么我们把累计和curSum再加上当前的数字,然后更新end位置为i。此时我们更新最大子数组之和mx,以及res即可,参见代码如下:

class Solution {
public:
    /**
     * @param A an integer array
     * @return  A list of integers includes the index of 
     *          the first number and the index of the last number
     */
    vector<int> continuousSubarraySum(vector<int>& A) {
        vector<int> res(2, -1);
        int curSum = 0, mx = INT_MIN, start = 0, end = 0;
        for (int i = 0; i < A.size(); ++i) {
            if (curSum < 0) {
                curSum = A[i];
                start = end = i;
            } else {
                curSum += A[i];
                end = i;
            }
            if (mx < curSum) {
                mx = curSum;
                res[0] = start;
                res[1] = end;
            }
        }
        return res;
    }
};

本文转自博客园Grandyang的博客,原文链接:连续子数组之和[LintCode] Continuous Subarray Sum ,如需转载请自行联系原博主。

相关文章
|
Java 索引 Python
LeetCode 209:最小长度的子数组 Minimum Size Subarray Sum
给定一个含有 n 个正整数的数组和一个正整数 s ,找出该数组中满足其和 ≥ s 的长度最小的连续子数组。如果不存在符合条件的连续子数组,返回 0。
550 0
|
算法
动态规划法(八)最大子数组问题(maximum subarray problem)
问题简介   本文将介绍计算机算法中的经典问题——最大子数组问题(maximum subarray problem)。所谓的最大子数组问题,指的是:给定一个数组A,寻找A的和最大的非空连续子数组。
1734 0