您现在的位置是:首页 >技术杂谈 >代码随想录算法训练营第四十九天网站首页技术杂谈

代码随想录算法训练营第四十九天

zszq111 2023-07-08 16:00:02
简介代码随想录算法训练营第四十九天

代码随想录算法训练营第四十九天| 121. 买卖股票的最佳时机,122. 买卖股票的最佳时机 II

121. 买卖股票的最佳时机

题目链接:买卖股票的最佳时机

暴力解法

198 / 211 超时了

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        # dp[i] = i天获取的最大利润
        dp = [0]*(len(prices)+1)

        for i in range(len(prices)):
            for j in range((i+1),len(prices)):
                if prices[i] <= prices[j]:
                    #print(prices[i],prices[j],prices[i]-prices[j])
                    dp[i+1] = max(dp[i+1],prices[j]-prices[i])
        return max(dp)

贪心算法

找到最左的最小值。

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

动态规划

需要一项来记录有没有持有股票:
dp[i][0] 表示第i天持有股票
dp[i][1] 表示第i天没有持有股票
这里要注意的,这个买股票是一次性的,所以如果之前没有持有,现在要买,那么不是dp[i-1][1]-prices[i-1]而是直接-prices[i-1],因为已经没持有那段就是没有利润。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0]*2 for _ in range(len(prices)+1)]
        #dp[i][0] 第i天持有
        #dp[i][1] 第i天不持有
        # 第一天持有了 dp[1][0] = -7
        # 第一天没持有了 dp[1][1] = 0
        dp[1][0] = -prices[0]
        for i in range(2,len(prices)+1):
            # 第i天持有的最大值: 第i-1天也持有; 第i-1天没持有但买了
            # 因为是一次性的所以第i-1天没持有的最大利润其实还是0
            dp[i][0] = max(dp[i-1][0], -prices[i-1])
            # 第i天不持有的最大值: 第i-1天也没持有; 第i-1天持有但卖了
            dp[i][1] = max(dp[i-1][1], dp[i-1][0]+prices[i-1])
        return dp[-1][1]

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

题目链接:买卖股票的最佳时机 II
和121相对应的就是这题,这里买卖不是一次性的,所以没有持有的那一段,是有利润的,也就是dp[i-1][1]-prices[i-1]

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0]*2 for _ in range(len(prices)+1)]
        #dp[i][0] 第i天持有
        #dp[i][1] 第i天不持有
        # 第一天持有了 dp[1][0] = -7
        # 第一天没持有了 dp[1][1] = 0
        dp[1][0] = -prices[0]
        for i in range(2,len(prices)+1):
            # 第i天持有的最大值: 第i-1天也持有; 第i-1天没持有但买了
            # 因为是一次性的所以第i-1天没持有的最大利润其实还是0
            dp[i][0] = max(dp[i-1][0], dp[i-1][1]-prices[i-1])
            # 第i天不持有的最大值: 第i-1天也没持有; 第i-1天持有但卖了
            dp[i][1] = max(dp[i-1][1], dp[i-1][0]+prices[i-1])
        return dp[-1][1]
风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。