买卖股票（简单和中等）

定义2个dp数组
dp1为当天买入股票产生的收益
dp2为当天卖出股票产生的收益
整体收益 = 买入收益+卖出收益
注意，不能连续买，买->卖

① 只能买卖一次

"""
dp1[i]：
要么当天买：-prices[i]
要么当天不买：跟前一天的收益一致  dp1[i-1]
dp2[i]:
要么当天卖：当天股票价格prices[i]+前一天买入的收益dp1[i-1]
要么当天不卖：跟前一天的收益一致  dp2[i-1]
"""
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices) == 0 or len(prices) == 1:
            return 0
        # 股票必须买了以后要卖才会产生收益
        dp1 = [-prices[0]] * len(prices)  # 第i天买入的收益
        dp2 = [0] * len(prices)  # 第i天卖出的收益
        for i in range(1, len(prices)):
            # 买入的最大收益:
            # 今天不买，那么跟昨天的买入收益一致
            # 今天买了，那么买入的收益为 0-prices[i]
            dp1[i] = max(dp1[i - 1], 0 - prices[i])
            # 卖出的最大收益:
            # 今天不卖，那么跟昨天的卖出收益一致dp2[i-1]
            # 今天卖了，那么收益为当天的股价prices[i]+ 当天前面买入的最大收益dp1[i-1]
            dp2[i] = max(dp2[i - 1], prices[i] + dp1[i - 1])
        # print(dp1, dp2)
        return dp2[-1]

②买卖多次

"""
dp1[i]：
要么当天买：dp2[i-1] - prices[i]
要么当天不买：跟前一天的收益一致  dp1[i-1]
dp2[i]:
要么当天卖：当天股票价格prices[i]+前一天买入的收益dp1[i-1]
要么当天不卖：跟前一天的收益一致  dp2[i-1]
"""
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices) == 0 or len(prices) == 1:
            return 0
        # 股票必须买了以后要卖才会产生收益
        dp1 = [-prices[0]] * len(prices)  # 第i天买入的收益
        dp2 = [0] * len(prices)  # 第i天卖出的收益
        for i in range(1, len(prices)):
            # 买入的最大收益:
            # 今天不买，那么跟昨天的买入收益一致
            # 今天买了，因为不能连续买，那么买入的收益为前一天卖出的最大收益dp2[i-1]-prices[i]
            dp1[i] = max(dp1[i - 1], dp2[i - 1] - prices[i])
            # 卖出的最大收益:
            # 今天不卖，那么跟昨天的卖出收益一致dp2[i-1]
            # 今天卖了，那么收益为当天的股价prices[i]+ 当天前面买入的最大收益dp1[i-1]
            dp2[i] = max(dp2[i - 1], prices[i] + dp1[i - 1])
        # print(dp1, dp2)
        return dp2[-1]

③买卖多次，并且收交易手续费fee

"""
dp1[i]：
要么当天买：dp2[i-1] - prices[i]
要么当天不买：跟前一天的收益一致  dp1[i-1]
dp2[i]:
要么当天卖：当天股票价格prices[i]+前一天买入的收益dp1[i-1]-手续费fee
要么当天不卖：跟前一天的收益一致  dp2[i-1]
"""
class Solution:
    def maxProfit(self, prices: List[int], fee: int) -> int:
        if len(prices) == 0 or len(prices) == 1:
            return 0
        # 股票必须买了以后要卖才会产生收益
        dp1 = [-prices[0]] * len(prices)  # 第i天买入的收益
        dp2 = [0] * len(prices)  # 第i天卖出的收益
        for i in range(1, len(prices)):
            # 买入的最大收益:
            # 今天不买，那么跟昨天的买入收益一致
            # 今天买了，那么买入的收益为 0-prices[i]
            dp1[i] = max(dp1[i - 1], dp2[i - 1] - prices[i])
            # 卖出的最大收益:只有卖出才收手续费
            # 今天不卖，那么就没有手续费，那么跟昨天的卖出收益一致dp2[i-1]
            # 今天卖了，那么收益为当天的股价prices[i]+ 当天前面买入的最大收益dp1[i-1] - 手续费fee
            dp2[i] = max(dp2[i - 1], prices[i] + dp1[i - 1] - fee)
        return dp2[-1]

④买卖多次，卖出后一天的冷冻期

"""
dp1[i]：
要么当天买：dp2[i-2] - prices[i]
要么当天不买：跟前一天的收益一致  dp1[i-1]
dp2[i]:
要么当天卖：当天股票价格prices[i]+前一天买入的收益dp1[i-1]
要么当天不卖：跟前一天的收益一致  dp2[i-1]
"""
class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices) == 0 or len(prices) == 1:
            return 0
        # 股票必须买了以后要卖才会产生收益，即收益=买入收益+卖出收益
        dp1 = [-prices[0]] * len(prices)  # 第i天买入的收益
        dp2 = [0] * len(prices)  # 第i天卖出的收益
        for i in range(1, len(prices)):
            # 买入的最大收益:
            # 今天不买，那么跟昨天的买入收益一致
            # 今天买了，因为含有冷冻期1天，冷冻期不能买卖，那么买入的收益为前2天卖出的最大收益dp2[i-2]-prices[i]
            dp1[i] = max(dp1[i - 1], dp2[i - 2] - prices[i])
            # 卖出的最大收益:
            # 今天不卖，那么跟昨天的卖出收益一致dp2[i-1]
            # 今天卖了，那么收益为当天的股价prices[i]+ 当天前面买入的最大收益dp1[i-1]
            dp2[i] = max(dp2[i - 1], prices[i] + dp1[i - 1])
        # print(dp1, dp2)
        return dp2[-1]