53. 最大子序和(剑指 Offer 42)

53. 最大子序和(剑指 Offer 42)

知识点:数组;前缀和;哨兵;动态规划;贪心;分治

题目描述

输入一个整型数组,数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。

要求时间复杂度为O(n)。

示例
输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。

解法一:前缀和+哨兵

连续子数组 –> 前缀和
从前往后遍历求前缀和,维持两个变量,一个是最大子数组和,也就是答案,一个是最小的前缀和,我们可以把这个值理解为哨兵,这个就是我们用来获取答案的,因为每次前缀和-这个最小的肯定就是最大的。

class Solution {
    public int maxSubArray(int[] nums) {
        int pre = 0; //前缀和;
        int minPre = 0; //最小的前缀和:哨兵;
        int maxSum = Integer.MIN_VALUE;
        for(int i = 0; i < nums.length; i++){
            pre += nums[i];
            maxSum = Math.max(maxSum, pre-minPre);
            minPre = Math.min(pre, minPre);
        }
        return maxSum;
    }
}

解法二:贪心

这道题贪心怎么解?贪什么呢?想一下在这个过程中,比如-2 1,我们需要-2吗?不需要!因为负数只会拉低我们最后的和,只起副作用的索性不如不要了。直接从1开始就行了; 贪的就是负数和一定会拉低结果。
所以我们的贪心选择策略就是:只选择和>0的,对于和<=0的都可以舍弃了。

class Solution {
    public int maxSubArray(int[] nums) {
        int maxSum = Integer.MIN_VALUE;
        int sum = 0;
        for(int i = 0; i < nums.length; i++){
            sum += nums[i];
            maxSum = Math.max(sum, maxSum);
            if(sum <= 0){
                sum = 0; //对于<=0的前缀和,已经没要意义了,从下一位置开始;
                continue;
            }
        }
        return maxSum;
    }
}

解法三:分治

这道题可以用分治去解。期望去求解一个区间[l,r]内的最大子序和,按照分而治之的思想,可以将其分为左区间和右区间。
左区间L:[l, mid]和右区间R:[mid + 1, r].
lSum 表示 [l,r] 内以 l 为左端点的最大子段和
rSum 表示 [l,r] 内以 r 为右端点的最大子段和
mSum 表示 [l,r] 内的最大子段和
iSum 表示 [l,r] 的区间和
递归地求解出L.mSum以及R.mSum之后求解M.mSum。因此首先在分治的递归过程中需要维护区间最大连续子列和mSum这个信息。
接下来分析如何维护M.mSum。具体来说有3种可能:

  • M上的最大连续子列和序列完全在L中,即M.mSum = L.mSum
  • M上的最大连续子列和序列完全在R中,即M.mSum = R.mSum
  • M上的最大连续子列和序列横跨L和R,则该序列一定是从L中的某一位置开始延续到mid(L的右边界),然后从mid + 1(R的左边界)开始延续到R中的某一位置。因此我们还需要维护区间左边界开始的最大连续子列和leftSum以及区间右边界结束的最大连续子列和rightSum信息
class Solution {
    public class Status{
        public int lSum, rSum, mSum, iSum;
        // lSum 表示 [l,r] 内以 l 为左端点的最大子段和
        // rSum 表示 [l,r] 内以 r 为右端点的最大子段和
        // mSum 表示 [l,r] 内的最大子段和
        // iSum 表示 [l,r] 的区间和
        public Status(int lSum, int rSum, int mSum, int iSum){
            this.lSum = lSum;
            this.rSum = rSum;
            this.mSum = mSum;
            this.iSum = iSum;
        }
    }
    public Status getInfo(int[] a, int l, int r){
        if(l == r) return new Status(a[l], a[l], a[l], a[l]); //终止条件;
        int mid = l + ((r-l) >> 1);
        Status lsub = getInfo(a, l, mid);
        Status rsub = getInfo(a, mid+1, r);
        return pushUp(lsub, rsub); 
    }
    //根据两个子串得到整个序列结果;
    public Status pushUp(Status l, Status r){
        int iSum = l.iSum + r.iSum;
        int lSum = Math.max(l.lSum, l.iSum+r.lSum);
        int rSum = Math.max(r.rSum, r.iSum+l.rSum);
        int mSum = Math.max(Math.max(l.mSum, r.mSum), l.rSum+r.lSum);
        return new Status(lSum, rSum, mSum, iSum);
    }

    public int maxSubArray(int[] nums) {
        return getInfo(nums, 0, nums.length-1).mSum;
    }
}

解法四:动态规划

  • 1.确定dp数组和其下标的含义:dp[i]表示以i结尾的连续子数组的最大和;
  • 2.确定递推公式,即状态转移方程:以i结尾想一下我们有几种可能,一种是i-1过来的,也就是上一个的连续子数组延续到i处了,那和就为dp[i-1]+nums[i],另一种呢,就是自己开始,前面那个连续子数组不行,那就是nums[i]了,想一下为什么前面那个不行,还不是前面的和会拖累自己,那就意味着前面的和是负数;这其实就引出贪心的方法了。不过我们这里不用这么麻烦,直接用一个max函数,取两者大的那个就行;
  • 3.dp初始化base case:dp[0]只有一个数,所以dp[0] = nums[0];
class Solution {
    public int maxSubArray(int[] nums) {
        int len = nums.length;
        int[] dp = new int[len]; //以i结尾的连续子数组的最大和为dp[i];
        if(nums == null || len <= 1) return nums[0];
        dp[0] = nums[0];
        for(int i = 1; i < len; i++){
            dp[i] = Math.max(dp[i-1]+nums[i], nums[i]); //状态转移;
        }
        //注意我们要遍历一遍返回最大的dp;
        int maxSum = dp[0];
        for(int i = 1; i < len; i++){
            maxSum = Math.max(maxSum, dp[i]);
        }
        return maxSum;
    }
}

当然上述程序可以优化,因为我们的dp[i]其实只和前一状态i-1有关,所以可以采用一个滚动变量来记录,而不用整个数组。

class Solution {
    public int maxSubArray(int[] nums) {
        int pre = 0;  //记录前一状态;
        int res = nums[0]; //记录最后结果的最大值;
        for (int num : nums) {
            pre = Math.max(pre + num, num);
            res = Math.max(res, pre);
        }
        return res;
    }
}

体会

这道题目是一道很典型的题目,用到了各种方法和思想。要常看常做,分治是其中比较困难的,但是要会这种思想。这道题目最好的方法还是哨兵和动态规划, 其实贪心就是从动态规划的一个特殊情况过去的,体会两者的关系;