腾讯47-买卖股票的最佳时机leetcode121
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。
注意你不能在买入股票前卖出股票。
示例 1:
输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。
示例 2:
输入: [7,6,4,3,1]
输出: 0
解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。
分析:
- 这里股票要求,先买后卖
- 买卖两个变量,如果用dp[i][j],就是o(n*n),等同于暴力,不可取
- 因为股票迟早要卖的,即使最后结果是0,相当于不卖,可认为是第0天卖的
- 所以用dp[i]表示如果第i天卖股票,能获得的最大收益,dp[0]=0,第一天没收入的,从dp[1]开始才有意义
- dp[i]表示如果第i天卖股票,肯定要减去之前的最小值,所有用一个min_cur,表示截止第i天,包括第i天的最小值
- 综上,res就是dp[i]中的最大值
class Solution:
def maxProfit(self, prices: List[int]) -> int:
#这里股票要求,先买后卖
#买卖两个变量,如果用dp[i][j],就是o(n*n),等同于暴力
#因为股票迟早要卖的,即使最后结果是0,相当于不卖,可认为是第0天卖的
#所以用dp[i]表示如果第i天卖股票,能获得的最大收益,dp[0]=0,第一天没收入的,从dp[1]开始才有意义
#dp[i]表示如果第i天卖股票,肯定要减去之前的最小值,所有用一个min_cur,表示截止第i天,包括第i天的最小值
#综上,res就是dp[i]中的最大值
if len(prices)==0 or len(prices)==1:return 0
else:
res=0
min_cur,dp=prices[0],[0]
for i in range(1,len(prices)):
min_cur=min(min_cur,prices[i])
dp.append(prices[i]-min_cur if prices[i]-min_cur>0 else 0)
res=max(res,dp[-1])
return res