如何用leetcode計算出買賣股票的最佳時機

本篇內容主要講解“如何用leetcode計算出買賣股票的最佳時機”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“如何用leetcode計算出買賣股票的最佳時機”吧!

一、題目內容

給定一個整數數組 prices ，它的第 i 個元素 prices[i] 是一支給定的股票在第 i 天的價格。

設計一個算法來計算你所能獲取的最大利潤。你最多可以完成 k 筆交易。

注意：你不能同時參與多筆交易（你必須在再次購買前出售掉之前的股票）。

示例 1：

輸入：k = 2, prices = [2,4,1]
輸出：2
解釋：在第 1 天 (股票價格 = 2) 的時候買入，在第 2 天 (股票價格 = 4) 的時候賣出，這筆交易所能獲得利潤 = 4-2 = 2 。

示例 2：

輸入：k = 2, prices = [3,2,6,5,0,3]
輸出：7
解釋：在第 2 天 (股票價格 = 2) 的時候買入，在第 3 天 (股票價格 = 6) 的時候賣出, 這筆交易所能獲得利潤 = 6-2 = 4 。
     隨后，在第 5 天 (股票價格 = 0) 的時候買入，在第 6 天 (股票價格 = 3) 的時候賣出, 這筆交易所能獲得利潤 = 3-0 = 3 。

提示：

0 <= k <= 109
0 <= prices.length <= 1000
0 <= prices[i] <= 1000

二、解題思路

動態規劃，每個price都需要檢查一遍；

然后買入和賣出的兩個狀態分別進行max比較即可。

三、代碼

import sys
class Solution:
    def maxProfit(self, k: int, prices: list) -> int:
        dp = [[0, 0] for _ in range(k + 1)]
        for j in range(k + 1):
            dp[j][1] = -sys.maxsize
        for i in range(0, len(prices)):
            for j in range(1, k + 1):
                dp[j][0] = max(dp[j][0], dp[j][1] + prices[i])  # 賣出
                dp[j][1] = max(dp[j][1], dp[j - 1][0] - prices[i])  # 買入
        return dp[k][0]


if __name__ == '__main__':
    k = 2
    prices = [3, 2, 6, 5, 0, 3]
    s = Solution()
    ans = s.maxProfit(k, prices)
    print(ans)

到此，相信大家對“如何用leetcode計算出買賣股票的最佳時機”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！