LintCode-最大子數(shù)組I、II、III

思路概述

I、非常常規(guī)的動態(tài)規(guī)劃問題,遞推公式就是dp[i] = max(nums[i], nums[i] + dp[i-1]), 在當(dāng)前i位置判斷是否舍棄i及i前面的加和結(jié)果,比當(dāng)前數(shù)字優(yōu)就不舍棄, 即dp[i-1]得大于零,也就是歷史結(jié)果得大于零才有延續(xù)下去的價(jià)值。
II、兩個(gè)數(shù)組相互無法重疊,必定是一個(gè)在左,一個(gè)在右,那么從正反兩個(gè)方向求出I中的dp_lift[i][0]、dp_right[i][0]記錄從左、從右必定加上當(dāng)前值情況下的局部最優(yōu)解,dp_lift[i][1]、dp_right[i][1]記錄從左、從右無論當(dāng)前值加不加情況下的全局最優(yōu)解。最后求出

max(dp_lift[i][1]+dp_right[i][1])

即可。
III、特殊動態(tài)規(guī)劃問題,Local[i][j]表示在到達(dá)數(shù)組中第i個(gè)數(shù)時(shí), 有j個(gè)子數(shù)組, 并且第i個(gè)數(shù)一定在子數(shù)組中,此為局部最優(yōu)。Global[i][j]為在到達(dá)數(shù)組中第i個(gè)數(shù)時(shí), 有j個(gè)子數(shù)組, 第i個(gè)數(shù)不一定在子數(shù)組中,此為全局最優(yōu)。

Local[i][j] = max(Local[i-1][j], Global[i-1][j-1]) + nums[i-1]=max(Local[i-1][j]+ nums[i-1], Global[i-1][j-1]+ nums[i-1])

Local[i-1][j]+ nums[i-1]表示把第i個(gè)數(shù)加入到有著第i-1個(gè)數(shù)的第j個(gè)子數(shù)組,Global[i-1][j-1]+ nums[i-1]表示用第i個(gè)數(shù)新產(chǎn)生第j個(gè)子數(shù)組,max(Local[i-1][j]+ nums[i-1], Global[i-1][j-1]+ nums[i-1]) 表示對這種兩種選擇狀態(tài)進(jìn)行擇優(yōu)。

Global[i][j] = max(Local[i][j], Global[i-1][j])

特別注意,遞推公式的max函數(shù)實(shí)際上代表了對過去狀態(tài)產(chǎn)生的多個(gè)結(jié)果進(jìn)行擇優(yōu)。

I

描述

給定一個(gè)整數(shù)數(shù)組,找到一個(gè)具有最大和的子數(shù)組,返回其最大和。
子數(shù)組最少包含一個(gè)數(shù)。

樣例

給出數(shù)組[?2,2,?3,4,?1,2,1,?5,3],符合要求的子數(shù)組為[4,?1,2,1],其最大和為6

代碼

class Solution:
    """
    @param nums: A list of integers
    @return: A integer indicate the sum of max subarray
    """
    def maxSubArray(self, nums):
        # write your code here
        dp = [0 for i in range(len(nums))]
        dp[0] = nums[0]
        for i in range(1, len(nums)):
            dp[i] = max(nums[i], nums[i] + dp[i-1])
        return max(dp)

II

描述

給定一個(gè)整數(shù)數(shù)組,找出兩個(gè) 不重疊 子數(shù)組使得它們的和最大。
每個(gè)子數(shù)組的數(shù)字在數(shù)組中的位置應(yīng)該是連續(xù)的。
返回最大的和。
子數(shù)組最少包含一個(gè)數(shù)。

樣例

給出數(shù)組 [1, 3, -1, 2, -1, 2]
這兩個(gè)子數(shù)組分別為 [1, 3] 和 [2, -1, 2] 或者 [1, 3, -1, 2] 和 [2],它們的最大和都是 7

挑戰(zhàn)

要求時(shí)間復(fù)雜度為 O(n)

代碼

class Solution:
    """
    @param: nums: A list of integers
    @return: An integer denotes the sum of max two non-overlapping subarrays
    """

    def maxTwoSubArrays(self, nums):
        # write your code here
        dp_lift = [[0, 0] for i in range(len(nums))]
        dp_right = [[0, 0] for j in range(len(nums))]

        dp_lift[0][0] = nums[0]
        dp_lift[0][1] = nums[0]
        for i in range(1, len(nums)):
            dp_lift[i][0] = max(nums[i], nums[i] + dp_lift[i - 1][0])
            dp_lift[i][1] = max(dp_lift[i][0], dp_lift[i - 1][1])
        dp_right[-1][0] = nums[-1]
        dp_right[-1][1] = nums[-1]
        for i in reversed(range(0, len(nums) - 1)):
            dp_right[i][0] = max(nums[i], nums[i] + dp_right[i + 1][0])
            dp_right[i][1] = max(dp_right[i][0], dp_right[i + 1][1])
        result = -10000
        for i in range(0, len(dp_lift)-1):
            result = max(result, dp_lift[i][1] + dp_right[i+1][1])
        return result

III

描述

給定一個(gè)整數(shù)數(shù)組和一個(gè)整數(shù) k,找出 k 個(gè)不重疊子數(shù)組使得它們的和最大。每個(gè)子數(shù)組的數(shù)字在數(shù)組中的位置應(yīng)該是連續(xù)的。
子數(shù)組最少包含一個(gè)數(shù)。

樣例

給出數(shù)組 [-1,4,-2,3,-2,3] 以及 k = 2,返回 8

代碼

class Solution:
    """
    @param nums: A list of integers
    @param k: An integer denote to find k non-overlapping subarrays
    @return: An integer denote the sum of max k non-overlapping subarrays
    """
    def maxSubArray(self, nums, k):
        # write your code here
        minNum = min(0, min(nums) - 1)
        Local = [[0 for i in range(k + 1)] for i in range(len(nums) + 1)]
        Global = [[0 for i in range(k + 1)] for i in range(len(nums) + 1)]

        for j in range(1, k+1):
            Local[0][j] = minNum
            Global[0][j] = minNum
        for i in range(1, len(nums)+1):
            for j in range(1, k+1):
                if j > i:
                    Local[i][j] = minNum
                    Global[i][j] = minNum
                else:
                    Local[i][j] = max(Local[i-1][j], Global[i-1][j-1]) + nums[i-1]
                    Global[i][j] = max(Local[i][j], Global[i-1][j])
        return Global[-1][-1]

題目來源

https://www.lintcode.com/problem/maximum-subarray-i/description
https://www.lintcode.com/problem/maximum-subarray-ii/description
https://www.lintcode.com/problem/maximum-subarray-iii/description

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容