152. 乘积最大子数组
Title
給你一個整數數組 nums ,請你找出數組中乘積最大的連續子數組(該子數組中至少包含一個數字),并返回該子數組所對應的乘積。
示例 1:
輸入: [2,3,-2,4]
輸出: 6
解釋: 子數組 [2,3] 有最大乘積 6。
示例 2:
輸入: [-2,0,-1]
輸出: 0
解釋: 結果不能為 2, 因為 [-2,-1] 不是子數組。
Solve
動態規劃
如果當前位置是一個負數的話,那么我們希望在它前面一個位置的某一段的積也是個負數,并且盡可能小,負負得正。
如果當前位置是一個正數的話,那么我們希望它前面一個位置的某一段的積也是個正數,并且希望它盡可能大。
這樣,我們可以維護一個fmax(i)和fmin(i),分別表示以第i個元素結尾的最大(小)乘積的子數組的乘積。
fmax(i)=maxi=1n{fmax(i?1)?ai,fmin(i?1)?ai,ai}f_{max}(i)=max^{n}_{i=1}\{f_{max}(i-1)*a_i,f_{min}(i-1)*a_i,a_i\} fmax?(i)=maxi=1n?{fmax?(i?1)?ai?,fmin?(i?1)?ai?,ai?}fmin(i)=mini=1n{fmax(i?1)?ai,fmin(i?1)?ai,ai}f_{min}(i)=min^{n}_{i=1}\{f_{max}(i-1)*a_i,f_{min}(i-1)*a_i,a_i\} fmin?(i)=mini=1n?{fmax?(i?1)?ai?,fmin?(i?1)?ai?,ai?}
它代表第i個元素結尾的乘積最大子數組的乘積fmax(i),可以考慮吧ai加入第i-1個元素結尾的乘積最大或最小的子數組的乘積中,二者加上ai,三者取大。
第i個元素結尾的乘積最小子數組的乘積fmin(i)同理。
由于第i個狀態只和第i-1個狀態有關,根據【滾動數組】思想,可以只用fmax(i)和fmin(i)兩個變量來維護i-1時刻的狀態。
Code
class Solution:def maxProduct(self, nums: List[int]) -> int:maxF, minF, ans = nums[0], nums[0], nums[0]for i in range(1, len(nums)):mx, mn = maxF, minFmaxF = max(mx * nums[i], max(mn * nums[i], nums[i]))minF = min(mn * nums[i], min(mx * nums[i], nums[i]))ans = max(ans, maxF)return ans總結
以上是生活随笔為你收集整理的152. 乘积最大子数组的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 210. 课程表 II
- 下一篇: 680. Valid Palindrom