【Lintcode】402. Continuous Subarray Sum

题目地址：

https://www.lintcode.com/problem/continuous-subarray-sum/description

给定一个数组，求其最大和连续子列，返回区间起点和终点下标。若有若干不同区间和都是最大，则返回字典序最小的那两个下标组合。

回顾最大和连续子列不需要返回端点下标，只需返回和，这一问题的做法。若以$f[i]$表示以$A[i]$结尾的连续子数组中的最大和，那么显然$$f[i+1]=\max (f[i]+A[i+1],A[i+1])$$这里并不需要用一个数组来记录$f[i]$，只需要用一个变量记录即可，随着遍历数组更新之。此外我们需要时时记录sum对应的区间两端点，同时发现更大的和的时候，把这两个端点更新到结果了即可。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    /*
     * @param A: An integer array
     * @return: A list of integers includes the index of the first number and the index of the last number
     */
    public List<Integer> continuousSubarraySum(int[] A) {
        // write your code here
        List<Integer> res = new ArrayList<>();
        
        if (A == null || A.length == 0) {
            return res;
        }
        res.add(0);
        res.add(0);
        // start和end维护当前sum对应区间的两个端点，
        // sum维护以A[i]结尾的最大子区间的和，maxSum维护已经发现的最大子区间和
        int start = 0, end = 0;
        int sum = A[0], maxSum = A[0];
        for (int i = 1; i < A.length; i++) {
        	// sum >= 0说明以A[i]结尾的最大和要将前面的sum连上去；
        	// 否则说明以A[i]结尾的最大和的区间就是A[i]自己
        	// 注意，如果sum = 0则说明A[i]可以连上前面的区间也可以不连，但为了字典序较小，应该选择不连
        	// start和end时时刻刻维护sum对应区间的两个端点
            if (sum >= 0) {
                sum += A[i];
                end = i;
            } else {
                sum = A[i];
                start = end = i;
            }
            
            // 发现更大的和，更新已发现的最大和并更新端点
            if (sum > maxSum) {
                maxSum = sum;
                res.set(0, start);
                res.set(1, end);
            }
        }
        
        return res;
    }
}

时间复杂度$O(n)$，空间$O(1)$。

关于为什么start和end维护了sum对应区间的两个端点，可以由数学归纳法得到。此处省略证明。