代码随想录 Day - 49｜#121 买卖股票的最佳时机｜#213 买卖股票的最佳时机 II

清单

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

LeetCode #121 买卖股票的最佳时机

1. 题目

给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择某一天买入这只股票，并选择在未来的某一个不同的日子卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0。

2. 思路

1. dp数组: 尝试一维数组 dp[i] 不能完整表达信息，因此采用二维数组dp[i][j]。其中i代表天数，j[0] 代表持有股票所得最大现金，j[1]代表卖出了股票所得最大现金。
2. 递推公式: 将买入股票定义为-prices[i]，则对于dp[i][0]，需要记录最大持有股票所得现金—> dp[i][0] = max(dp[i-1], -prices[i])。对于dp[i][1]，需要记录卖出股票后的所得最大现金 —> dp[i][1] = max(prices[i] + dp[i-1][0], dp[i-1][1])
3. 初始化: dp[0][0] = -prices[i], dp[0][1] = 0
4. 遍历顺序: 从前往后

3. 代码实现

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        #Initial dp
        dp = [[0] * 2 for _ in range(len(prices))]

        #Original Setting
        dp[0][0] = - prices[0]
        dp[0][1] = 0
        for i in range(1, len(prices)):
                dp[i][0] = max(-prices[i],dp[i-1][0])
                dp[i][1] = max(dp[i-1][1],prices[i] + dp[i-1][0])
        
        return dp[-1][1]

LeetCode #213 买卖股票的最佳时机 II

1. 题目

给你一个整数数组 prices，其中 prices[i] 表示某支股票第 i 天的价格。
在每一天，你可以决定是否购买和/或出售股票。你在任何时候最多只能持有一股股票。你也可以先购买，然后在同一天出售。
返回你能获得的最大利润。

2. 思路

1. dp数组含义: dp[i][0] 第i天持有股票的最大金额，dp[i][1] 第i天卖出股票持有的最大金额
2. 递推公式: 因为可以在当天出售 因此 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])
3. 初始化: dp[0][0] = - prices[0] dp[0][1] = 0
4. 遍历顺序: 从前往后

3. 代码实现

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        dp = [[0] * 2 for _ in range(len(prices))]
        dp[0][0] = - prices[0]
        dp[0][1] = 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])
        
        return dp[-1][1]