玩命加载中 . . .

最大子数组和


描述

给你一个整数数组 nums ,请你找出一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。

子数组 是数组中的一个连续部分。

示例 1:

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

示例 2:

输入:nums = [1]
输出:1

示例 3:

输入:nums = [5,4,-1,7,8]
输出:23

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/maximum-subarray
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

这个问题可以使用动态规划来解决。我们可以定义一个变量 curSum 来表示当前的子序列和,以及一个变量 maxSum 来表示最大子序列和。我们遍历整个数组,将每个元素加到 curSum 中,并在每次加入元素后更新 maxSum 的值。如果 curSum 变成了负数,我们就将其重置为0,因为负的子序列和不可能是最大子序列和的一部分。

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        int maxSum = nums[0]; // 初始化最大和为数组第一个元素
        int curSum = 0; // 初始化当前和为0
        for (int i = 0; i < nums.size(); i++) {
            curSum += nums[i]; // 累加当前元素到当前和
            maxSum = max(maxSum, curSum); // 更新最大和
            if (curSum < 0) { // 如果当前和小于0,重置当前和为0
                curSum = 0;
            }
        }
        return maxSum; // 返回最大和
    }
};

题目还要求我们用分治法来解,这个分治法的思想就类似于二分搜索法,需要把数组一分为二,分别找出左边和右边的最大子数组之和,然后还要从中间开始向左右分别扫描,求出的最大值分别和左右两边得出的最大值相比较取最大的那一个,代码如下:

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        if (nums.empty()) return 0;
        return helper(nums, 0, (int)nums.size() - 1);
    }
    int helper(vector<int>& nums, int left, int right) {
        if (left >= right) return nums[left];
        int mid = left + (right - left) / 2;
        int lmax = helper(nums, left, mid - 1);
        int rmax = helper(nums, mid + 1, right);
        int mmax = nums[mid], t = mmax;
        for (int i = mid - 1; i >= left; --i) {
            t += nums[i];
            mmax = max(mmax, t);
        }
        t = mmax;
        for (int i = mid + 1; i <= right; ++i) {
            t += nums[i];
            mmax = max(mmax, t);
        }
        return max(mmax, max(lmax, rmax));
    }
};

回顾一下,前缀和数组 preSum 就是 nums 元素的累加和,preSum[i+1] - preSum[j] 其实就是子数组 nums[j..i] 之和(根据 preSum 数组的实现,索引 0 是占位符,所以 i 有一位索引偏移)。

**那么反过来想,以 nums[i] 为结尾的最大子数组之和是多少?其实就是 preSum[i+1] - min(preSum[0..i])**。

所以,我们可以利用前缀和数组计算以每个元素结尾的子数组之和,进而得到和最大的子数组:

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        int n = nums.size();
        vector<int> preSum(n + 1);
        preSum[0] = 0;
        // 构造 nums 的前缀和数组
        for (int i = 1; i <= n; i++) {
            preSum[i] = preSum[i - 1] + nums[i - 1];
        }
    
        int res = INT_MIN;
        int minVal = INT_MAX;
        for (int i = 0; i < n; i++) {
            // 维护 minVal 是 preSum[0..i] 的最小值
            minVal = min(minVal, preSum[i]);
            // 以 nums[i] 结尾的最大子数组和就是 preSum[i+1] - min(preSum[0..i])
            res = max(res, preSum[i + 1] - minVal);
        }
        return res;
    }
};

文章作者: Jack Tim
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 Jack Tim !
评论
  目录