leetcode(5)股票买卖系列题目

动态规划:

(1)121. 买卖股票的最佳时机

注意:只能买卖一次

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        n = len(prices)
        dp = [[0] * 2 for _ in range(n)]
        dp[0][0] = -prices[0]  # 持有
        for i in range(1, n):
            dp[i][0] = max(dp[i - 1][0], - prices[i])
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])
        return dp[-1][1]

(2)122. 买卖股票的最佳时机 II

注意:买卖次数不限

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        n = len(prices)
        dp = [[0] * 2 for _ in range(n)]
        dp[0][0] = -prices[0]  # 持有
        for i in range(1, n):
            dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i])
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])
        return dp[-1][1]

(3)123. 买卖股票的最佳时机 III

注意:最多可以交易两次,因此可以买卖一次,可以买卖两次,也可以不买卖,一天一共就有五个状态:

  1. 没有操作dp[i][0] = dp[i - 1][0]
  2. 第一次买入dp[i][1] = max(dp[i - 1][1], dp[i-1][0] - prices[i]);
  3. 第一次卖出dp[i][2] = max(dp[i - 1][2], dp[i - 1][1] + prices[i]);
  4. 第二次买入dp[i][3] = max(dp[i - 1][3], dp[i - 1][2] - prices[i]);
  5. 第二次卖出dp[i][4] = max(dp[i - 1][4], dp[i - 1][3] + prices[i]);

dp[i][j]中 i表示第i天,j为 [0 - 4] 五个状态,dp[i][j]表示第i天状态j所剩最大现金

利润最大的时候一定是卖出的状态,而两次卖出的状态现金最大一定是最后一次卖出。

所以最终最大利润是dp[-1][4]

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0] * 5 for _ in range(len(prices))]
        dp[0][1] = dp[0][3] = -prices[0]  # 持有
        for i in range(1, len(prices)):
            dp[i][0] = dp[i - 1][0]
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] - prices[i])
            dp[i][2] = max(dp[i - 1][2], dp[i - 1][1] + prices[i])
            dp[i][3] = max(dp[i - 1][3], dp[i - 1][2] - prices[i])
            dp[i][4] = max(dp[i - 1][4], dp[i - 1][3] + prices[i])
        return dp[-1][4]

(4)188. 买卖股票的最佳时机 IV

注意:最多可以交易k次,那么j的范围就定义为 2 * k + 1 就可以了。除了0以外,奇数就是买入,偶数就是卖出。

class Solution:
    def maxProfit(self, k: int, prices: List[int]) -> int:
        n = len(prices)
        if n == 0:return 0
        dp = [[0] * (2 * k + 1) for _ in range(n)]
        for i in range(1, 2 * k, 2):  # 持有
            dp[0][i] = -prices[0]
        for i in range(1, n):
            for j in range(1, 2 * k, 2):
                dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - 1] - prices[i])
                dp[i][j + 1] = max(dp[i - 1][j + 1], dp[i - 1][j] + prices[i])
        return dp[-1][-1]

(5)309. 最佳买卖股票时机含冷冻期

注意:买卖次数不限,但卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。

  • 状态一:买入股票状态(今天买入股票,或者是之前就买入了股票然后没有操作)
  • 卖出股票状态,这里就有两种卖出股票状态
    • 状态二:两天前就卖出了股票,度过了冷冻期,一直没操作,今天保持卖出股票状态
    • 状态三:今天卖出了股票
  • 状态四:今天为冷冻期状态,但冷冻期状态不可持续,只有一天!
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0] * 4 for _ in range(len(prices))]
        dp[0][0] = -prices[0]  # 持有
        for i in range(1, len(prices)):
            # 今天持有 = max(前一天买入,max(前一天保持卖出的状态,前一天是冷冻期)且今天买入)
            dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][1], dp[i - 1][3]) - prices[i])
            # 今天保持卖出 = max(前一天卖出,前一天是冷冻期)
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][3])  #后面一种情况应该是持续冷冻期状态了吧?
            # 今天卖出 = 前一天买入 + 今日价格
            dp[i][2] = dp[i - 1][0] + prices[i]
            # 今天冷冻期
            dp[i][3] = dp[i - 1][2]
        return max(dp[-1][1], dp[-1][2], dp[-1][3])

(6)714. 买卖股票的最佳时机含手续费

注意:要统一在卖出的时候减手续费

class Solution:
    def maxProfit(self, prices: List[int], fee: int) -> int:
        dp = [[0] * 2 for _ in range(len(prices))]
        dp[0][0] = -prices[0]  # 持有
        for i in range(1, len(prices)):
            dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i])
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i] - fee)
        return dp[-1][1]

贪心算法:

(1)121. 买卖股票的最佳时机

减左边最小的

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        n = len(prices)
        res = 0 
        low = prices[0]
        for i in range(n):
            low = min(low, prices[i])
            res = max(res, prices[i] - low)
        return res

(2)122. 买卖股票的最佳时机 II

每次累计大于零的利润

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        res = 0
        for i in range(1, len(prices)):
            res += max(0, prices[i] - prices[i - 1])
        return res

(6)714. 买卖股票的最佳时机含手续费

class Solution:
    def maxProfit(self, prices: List[int], fee: int) -> int:
        res = 0
        min_pri = prices[0]
        for i in range(1, len(prices)):
            if prices[i] < min_pri:
                # 买入,但是没有算手续费
                min_pri = prices[i]
            elif prices[i] >= min_pri and prices[i] <= min_pri + fee:
                # 两边都有=
                # 保持上一个状态,不买也不卖
                continue
            else:
                # 卖出,在此时算手续费,因为一笔交易可能买卖多次但只算一次手续费
                res += prices[i] - min_pri - fee
                min_pri = prices[i] - fee  # 注意-fee,下一次计算利润就不会重复出手续费
        return res

参考资料:
股票问题总结篇

posted @ 2022-05-11 14:55  YTT77  阅读(43)  评论(0编辑  收藏  举报