Q:
给定一个数组,它的第 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。
A:
1.DP:
思路:dp[i]记录第i天卖出的最大利润,则利润要么等于昨天买今天卖,要么今天买今天卖,如果dp[i-1]>0,说明之前某天买昨天卖是赚钱的,那么再试试之前某天买今天卖,取最大值。
class Solution:
def maxProfit(self, prices: List[int]) -> int:
l=len(prices)
if l<2:
return 0
dp=[0 for i in range(l+1)]
#dp[i]表示i-1天卖出的最大利润
dp[0]=0
dp[1]=0
res=0
for i in range(2,l+1):
dp[i]=max(prices[i-1]-prices[i-2],0)
if dp[i-1]>0:
dp[i]=max(dp[i-1]+prices[i-1]-prices[i-2],dp[i])
res=max(res,dp[i])
return res
2.遍历,和DP一样都是O(N)时间。
思路:最高利润出现在:买入为价格最低时,卖出为买入之后价格最高时。
故当更新最小值价格时,之前的最大价格要舍弃,从当前索引继续考察。
class Solution:
def maxProfit(self, prices: List[int]) -> int:
l=len(prices)
if l<2:
return 0
_min,_max=0,0
i=0
res=0
while i<l:
if prices[i]<prices[_min]:
_min=i
_max=i
if prices[i]>prices[_max]:
_max=i
res=max(res,prices[_max]-prices[_min])
i+=1
return res