53. 最大子数组和(中)

发布时间 2023-11-24 10:19:33作者: Frommoon

题目

  • 给你一个整数数组 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

动态规划

def maxSubArray(nums):
    n = len(nums)
    if n == 0:
        return 0  # 空数组的最大子数组和为0
    dp = [0] * n  # 创建一个长度为n的数组来保存动态规划的状态
    dp[0] = nums[0]  # 第一个元素作为初始值

    for i in range(1, n):
        dp[i] = max(nums[i], nums[i] + dp[i - 1])#计算以索引 i 结尾的子数组的最大和,即 dp[i]。这个值可以通过比较当前元素 nums[i] 和当前元素与前一个子数组的和 nums[i] + dp[i - 1] 的大小,取较大值作为 dp[i] 的值。

    res = float('-inf')  # 初始化结果为负无穷
    for i in range(n):
        res = max(res, dp[i])  # 更新最大和的结果

    return res

优化

def maxSubArray(nums):
    n = len(nums)
    if n == 0:
        return 0  # 空数组的最大子数组和为0
    dp_0 = nums[0]  # 初始化动态规划的状态,第一个元素作为初始值
    dp_1 = 0
    res = dp_0

    for i in range(1, n):
        # dp[i] = max(nums[i], nums[i] + dp[i-1])
        dp_1 = max(nums[i], nums[i] + dp_0)
        dp_0 = dp_1
        # 同时计算最大的结果
        res = max(res, dp_1)

    return res