LeetCode 第 53 題:連續(xù)子數組的最大和

標簽(空格分隔): 動態(tài)規(guī)劃 分治法

傳送門:53. 最大子序和。

給定一個整數數組 nums ,找到一個具有最大和的連續(xù)子數組(子數組最少包含一個元素),返回其最大和。

示例:

輸入: [-2,1,-3,4,-1,2,1,-5,4],
輸出: 6
解釋: 連續(xù)子數組 [4,-1,2,1] 的和最大,為 6。

進階:

如果你已經實現復雜度為 O(n) 的解法,嘗試使用更為精妙的分治法求解。

分析:


總結:分類討論的標準是:若之前的和小于 0,則將最大和置為當前值,否則計算最大和。

思路1:動態(tài)規(guī)劃

下面展示了標準的動態(tài)規(guī)劃的寫法。

Java 代碼:

public class Solution {
    
    public int FindGreatestSumOfSubArray(int[] array) {
        int n = array.length;
        if (n == 0) {
            return 0;
        }
        int[] dp = new int[n];
        dp[0] = array[0];
        int res = array[0];
        for (int i = 1; i < n; i++) {
            dp[i] = Integer.max(dp[i - 1] + array[i], array[i]);
            res = Integer.max(res, dp[i]);
        }
        return res;
    }

    public static void main(String[] args) {
        int[] nums = new int[]{6, -3, -2, 7, -15, 1, 2, 2};
        Solution solution = new Solution();
        int findGreatestSumOfSubArray = solution.FindGreatestSumOfSubArray(nums);
        System.out.println(findGreatestSumOfSubArray);
    }
}

這道題主要就在狀態(tài)的定義上要思考一下,這里題目中的關鍵字是“連續(xù)”,所以如果我們定義的狀態(tài)就是題目要求的結果:dp[i] 表示 nums 在區(qū)間 [0,i] 中連續(xù)子數組的最大和,那么在思考狀態(tài)轉移方程的時候,dp[i] 之前的,例如 dp[i-1] 就有可能是是更前面的連續(xù)子數組的最大和,不利于我們分類討論。

因此,我們可以定義狀態(tài):dp[i] 表示以 nums[i] 結尾的連續(xù)子數組的最大和。

這樣定義狀態(tài),分類討論就變得容易多了,因為 dp[i-1] 表示一定以 nums[i-1] 結尾,那么 dp[i] 就可以有兩種情況:

1、把 nums[i] 直接接在 dp[i-1] 表示的那個數組的后面;

例如,dp[i-1] = 3,nums[i] = 5,當然接在后面,越接越大。

2、單獨的一個 nums[i] 。

這種情況也比較好想到,比如:dp[i-1] = -3,nums[i] = 5,加上前面的數反而我越來越小了,干脆我另起爐灶吧。

以上兩種情況的最大值就是 dp[i] 的值。

最后不要忘記了,最終的結果應該是把所有的 dp[0],dp[1],……,dp[n-1] 都看一遍,求最大值。

重點:動態(tài)規(guī)劃問題。狀態(tài)是:以當前數字為結尾的連續(xù)子數組的最大和。

Python 代碼:

class Solution(object):
    def maxSubArray(self, nums):
        """
            :type nums: List[int]
            :rtype: int
            """

        l = len(nums)
        if l == 0:
            return 0
        if l == 1:
            return nums[0]
        dp = [0 for _ in range(l)]
        dp[0] = nums[0]

        for i in range(1, l):
            dp[i] = max(dp[i - 1] + nums[i], nums[i])
            # 最后不要忘記拉通求一遍最大值,或者在上面遍歷的時候,就保存最大值
            return max(dp)

或者你可以在遍歷的時候,就把最大值求出來。

class Solution(object):
    def maxSubArray(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        l = len(nums)
        if l == 0:
            return 0
        # 以索引 i 結尾的最大子數組的和
        end_i_max = nums[0]
        # 最后返回的數
        res = nums[0]
        for i in range(1, l):
            # 例:[-3,1]
            end_i_max = max(nums[i], end_i_max + nums[i])
            res = max(res, end_i_max)
        return res

思路2:分治法

參考資料:連續(xù)子數組最大和

Python 寫法:

class Solution(object):
    def maxSubArray(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n = len(nums)
        if n == 0:
            return 0
        return self.__max_sub_array(nums, 0, n - 1)

    def __max_sub_array(self, nums, left, right):
        if left == right:
            return nums[left]
        mid = left + (right - left) // 2
        return max(self.__max_sub_array(nums, left, mid),
                   self.__max_sub_array(nums, mid + 1, right),
                   self.__max_cross_array(nums, left, mid, right))

    def __max_cross_array(self, nums, left, mid, right):
        """
        一定包含 nums[mid] 元素的最大連續(xù)子數組的和
        思路是看看左邊擴散到底,得到一個最大數
        右邊擴散到底得到一個最大數
        :param nums:
        :param mid:
        :param right:
        :return:
        """
        ls = 0
        j = mid - 1
        s1 = 0
        while j >= left:
            s1 += nums[j]
            ls = max(ls, s1)
            j -= 1

        rs = 0
        j = mid + 1
        s2 = 0
        while j <= right:
            s2 += nums[j]
            rs = max(rs, s2)
            j += 1

        return ls + nums[mid] + rs


if __name__ == '__main__':
    s = Solution()
    nums = [-2, 1, -3, 4, -1, 2, 1, -5, 4]
    result = s.maxSubArray(nums)
    print(result)

(本節(jié)完)

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

相關閱讀更多精彩內容

友情鏈接更多精彩內容