自学内容网 自学内容网

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

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

给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0 。
示例 1:
输入:[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
示例 2:
输入:prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 没有交易完成, 所以最大利润为 0。
提示:
1 <= prices.length <= 105
0 <= prices[i] <= 104

动态规划

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        # dp[i][0] 表示在第i天及以前买入股票最低的价格
        # dp[i][1] 表示在第i天卖出股票的收益

        # 有的时候交换维度会很方便
        # dp[0][i] 表示在第i天及以前买入股票最低的价格
        # dp[1][i] 表示在第i天卖出股票的收益
        # dp[0][i] = min(dp[0][i-1], prices[i])
        # dp[1][i] = prices[i] - dp[0][i]
        n = len(prices)
        dp = [[sys.maxsize] * n, [0] * n]
        dp[0][0] = prices[0]
        dp[1][0] = 0

        for i in range(1, n):
            dp[0][i] = min(dp[0][i - 1], prices[i])
            dp[1][i] = prices[i] - dp[0][i]
        return max(dp[1])

使用滚动数组思想优化


class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        min_price, profit = sys.maxsize, 0
        for price in prices:
            min_price = min(min_price, price)
            profit = max(profit, price - min_price)
        return profit


原文地址:https://blog.csdn.net/UZDW_/article/details/142737113

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!