LintCode:402. 连续子数组求和

描述

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

样例

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

分析:变量ans保存当前最大的连续子数组之和,sum表示当前start-end之间的和,遍历数组,如果sum<0,则重置start和end位置,其他情况,则只需要把end位置更新即可。然后sum与ans比较,更新ans,使得ans保存的始终是当前最大的连续子数组之和,把start和end位置也进行更新。

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
        ArrayList<Integer> arr=new ArrayList<>();
        arr.add(0); //开始位置
        arr.add(0); //结束位置
        int sum=0;
        int ans=Integer.MIN_VALUE;
        int start=0,end=0;
        for(int i=0;i<A.length;i++){
            if(sum<0){
                sum=A[i];
                start=end=i;
            }else{
                sum+=A[i];
                end=i;
            }
            if(sum>ans){
                ans=sum;
                arr.set(0,start);
                arr.set(1,end);
            }
        }
        return arr;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_27139155/article/details/81206540