代码随想录算法训练营第四十八天| LeetCode121. 买卖股票的最佳时机、122.买卖股票的最佳时机II、123.买卖股票的最佳时机III

121. 买卖股票的最佳时机

题目描述: 121. 买卖股票的最佳时机.

解法

dp

class Solution(object):
    def maxProfit(self, prices):
        if not prices:
            return 0
        dp0 = 0
        # 0表示不持有股票，1表示持有股票
        dp1 = 0-prices[0]
        for i in range(1,len(prices)):
            # 当前没有股票
            # 两种情况，前一天也没有，前一天的有但今天卖了
            dp0 = max(dp0,dp1+prices[i])
            # 当前有股票，昨天就有今天没卖，之前没有今天买了
            dp1 = max(dp1,0-prices[i])
        return dp0

这个题是只能买一次然后卖一次，只有一次机会。
可以使用两个变量来代替这个

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

题目描述: 122.买卖股票的最佳时机II.

解法

dp

class Solution(object):
    def maxProfit(self, prices):
        dp0 = 0
        dp1 = -prices[0]
        for i in range(len(prices)):
            tmp = dp0
            dp0 = max(dp0,dp1+prices[i])
            dp1 = max(dp1,tmp-prices[i])
        return dp0

这个是随时买随时卖。

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

题目描述: 123.买卖股票的最佳时机III.

解法

dp

class Solution(object):
    def maxProfit(self, prices):
        #0代表没有，1代表有第一支，2代表卖了第一支的没有，3代表有第二支，4代表第二支卖了。
        dp = [[0]*5 for _ in range(len(prices))]
        dp[0][1] = -prices[0]
        dp[0][2] = 0
        dp[0][3] = -prices[0]
        dp[0][4] = 0
        for i in range(1,len(prices)):
            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 max(dp[len(prices)-1][0],dp[len(prices)-1][2],dp[len(prices)-1][4])

用多维数组表示状态。