【leetcode】53. 最大子序和(Java)

题目描述

给定一个整数数组 nums ，找到一个具有最大和的连续子数组（子数组最少包含一个元素），返回其最大和。

题解

一、动态规划

• dp数组的含义: 以当前下标为结尾的最大连续子序列的和
• dp数组初始化：dp[0]，即以第一个元素为结尾的，最大连续子序列的和，即nums[0]
• dp递推公式：dp[i]，可以是以上一个结尾的子序列和 + 当前数字或者就是当前数字（前一个序列和起反作用了）

dp[i] = Math.max(nums[i], nums[i] + dp[i - 1]);

完整代码：

class Solution {
    public int maxSubArray(int[] nums) {
        int[] dp = new int[nums.length];
        dp[0] = nums[0];
        int res = nums[0];
        for (int i = 1; i < nums.length; i++){
            dp[i] = Math.max(nums[i], nums[i] + dp[i - 1]);
            res = Math.max(dp[i], res);
        }
        return res;
    }
}

二、贪心

• 维护当前和sum值，sum += 当前数字
• 比较res和sum
• 如果此时sum < 0， 就表示当前子序列对后面没有增益作用，所以sum置0，从下一位重新开始统计
• 要注意的有两点
  • res的初始值一定要设置为最小值，才可以保证不错过解
  • 比较res和sum，一定要在判断sum < 0之前进行。 如果顺序颠倒了的话，如果遇到全是负数的序列，sum就会一直被置0，然后再和res比较，所以就会得到错误的结果0

class Solution {
    public int maxSubArray(int[] nums) {
        int sum = 0, res = Integer.MIN_VALUE;
        for (int num : nums){
            sum += num;
            res = Math.max(res, sum);
            if (sum < 0) sum = 0;
        }
        return res;
    }
}

三、扩展题目
扩展题目要求返回最大子数组，而不是序列和。如:

输入：nums = [-2,1,-3,4,-1,2,1,-5,4]
输出： [4,-1,2,1]

同意利用贪心方法，在维护sum的同时维护两个变量left和right，用来标记出子数组的边界。详见代码注释

public class LC53 {
    public static void main(String[] args) {
        int[] nums = {-2,1,-3,4,-1,2,1,-5,4};
        int resLeft = 0, resRight = 0;
        int sum = 0, left = 0, right = 0, max = Integer.MIN_VALUE;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
            //更新右边界
            right = i;
            //如果此时sum > 当前最大值max。 更新max和结果的边界resLeft、resRight
            if (sum > max){
                max = sum;
                resRight = right;
                resLeft = left;
            }
            //如果sum < 0了，此时应该重新统计子序列，sum置0，left从下一个数字开始
            if (sum < 0){
                left = i + 1;
                sum = 0;
            }
        }
        //根据resleft和resRight构建结果数组
        int[] res = new int[resRight - resLeft + 1];
        for (int i = resLeft; i <= resRight; i++){
            res[i - resLeft] = nums[i];
        }
        //输出结果
        for (int re : res) {
            System.out.println(re);
        }
    }
}