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

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

难度:简单

题目描述
LeetCode 121.买卖股票的最佳时机(一)_第1张图片

思路求解

动态规划法求解,初始化一个表,保存每一天股票的涨跌,设为diff
其中diff[i] 表示第 i 天的股票与第 i-1 天的股票差价

初始化动态规划中保留子问题所用的表,设为dp
其中,dp[i] = max( 0, dp[i-1] + diff[i] )
然后记录最大的dp[i]

解释一下上面的最优子结构公式啊,这其实可以看成一个连续型的问题,但是我们不允许其中出现负值(即亏钱)。所以如果股票一直在涨,我们的收益便连续的涨,若某一天跌倒特别低了,我们不卖便是,相反由于股票价格低,我们考虑买入,然后再随着价格增长卖出。另外需要我们记录中间的差价。

代码求解

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if len(prices) < 2 : return 0
        diff = [0]
        pmax = 0
        for i in range(1,len(prices)):
            diff.append(prices[i]-prices[i-1])
        dp = [0]
        for i in range(1,len(prices)):
            dp.append(max(0,dp[i-1]+diff[i]))
            pmax = max(pmax, dp[i])
        
        return pmax

求解结果
LeetCode 121.买卖股票的最佳时机(一)_第2张图片

欢迎关注公众号 : 数学算法实验室

你可能感兴趣的:(LeetCode刷题日记)