leetcode_152. 乘积最大子数组

题目描述
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
示例 1:
输入: [2,3,-2,4]
输出: 6
解释: 子数组 [2,3] 有最大乘积 6。
示例 2:
输入: [-2,0,-1]
输出: 0
解释: 结果不能为 2, 因为 [-2,-1] 不是子数组。

分析: 动态规划
dp_max[i]:以第 i 个元素的结尾的连续子序列的最大乘积, 如果当前的第 i 个元素为正,可以直接乘,如果为负,就不乘,按照这个想法,最终的结果就是错误的,如 [-2,2,3,-3] 对应 dp = [-2, 2, 6 , -3] 正确结果为 [-2,2,6, -2 * 2 * 3 * -3]。
所以,我们应该考虑当前元素正负的情况,如果当前元素为正,nums[i] 应该 乘上 以 nums[i-1]结尾的尽可能大的正数,如果当前元素为负,nums[i] 应该乘上 以nums[i-1] 结尾的尽可能小的负数,这样才能更大的数。
于是,我们再维护一个 dp_min[i] ,以 nums[i] 结尾的连续子序列的最小乘积。
dp_max[i] = max(dp_max[i-1] * nums[i] ,dp_min[i-1] * nums[i-1] , nums[i])
dp_min[i] = min(dp_max[i-1] * nums[i],dp_min[i-1] * nums[i-1],nums[i])
base case :
dp_max[0] = nums[0]
dp_min[0] = nums[0]
最终结果:max(dp_max)
同理,我们求连续子序列的最小乘积也要维护 dp_max,dp_min 数组,最后只要求 min(dp_min)即可。

def maxProduct( nums):
    """
    :type nums: List[int]
    :rtype: int
    """
    n = len(nums)
    if n == 1 :
        return nums[0]
    dp_max = [0] * n  
    dp_min = [float('inf')] * n 
    dp_max[0] = max([0,nums[0]])
    dp_min[0] = nums[0]
    for i in range(1,n):
        dp_max[i] = max([dp_max[i-1] * nums[i],dp_min[i-1] * nums[i],nums[i]] )
        dp_min[i] = min([dp_max[i-1] * nums[i] ,dp_min[i-1] * nums[i],nums[i]])
    print(dp_min)
    return max(dp_max)             

优化空间:
由于第 i 个状态只和第 i - 1 个状态相关,根据「滚动数组」思想,我们可以只用两个变量来维护 i - 1 时刻的状态,一个维护 dp_max , 一个维护dp_min.
空间复杂度: O(1)

def maxProduct( nums):
    """
    :type nums: List[int]
    :rtype: int
    """
    n = len(nums)
    if n == 1 :
        return nums[0]
    max_f = nums[0]
    min_f = nums[0]
    ans = nums[0]
    
    for i in range(1,n):
        imax = max_f
        imin = min_f
        max_f = max([imax* nums[i],imin * nums[i],nums[i]] )
        print(max_f)
        min_f = min([imax * nums[i] ,imin * nums[i],nums[i]])
        ans = max(max_f,ans)
    return ans 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值