题目
假设把某股票的价格按照时间先后顺序存储在数组中,请问买卖该股票一次可能获得的最大利润是多少?
示例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。
思路
求第 i 天的最大利润dp[ i ],是求这两种情况中的最大值:一种是在第 i 天之前已经卖出,另一种是在第 i 天卖出,
如果在第 i 天之前已经卖出,那么第 i 天的最大利润dp[ i ]就是第 i-1 天的最大利润dp[ i-1 ];
如果在第 i 天卖出,那么第 i 天的最大利润dp[ i ]就是第 i 天的价格减去前 i-1 天的最低价格。
动态转移方程:
dp[i] = max(dp[i-1],prices[i]-min(prices[0:i]))
因为prices数组的下标是从0开始,所以,第 i 天的价格实际上是prices[ i -1 ]。
代码
class Solution:
def maxProfit(self, prices: List[int]) -> int:
n = len(prices)
if n==0 or n==1: return 0
dp = [0,0]
min_value = prices[0]
for i in range(2,n+1):
min_value = min(min_value,prices[i-2])
res = max(dp[i-1],prices[i-1]-min_value)
dp.append(res)
return dp[n]
class Solution:
def maxProfit(self, prices: List[int]) -> int:
if not prices: return 0
dp = [0 for _ in range(len(prices))]
min_cost = prices[0]
for i in range(1,len(prices)):
if prices[i-1]<min_cost:
min_cost = prices[i-1]
dp[i] = max(dp[i-1],prices[i]-min_cost)
return dp[-1]
复杂度
时间复杂度 O(N)
空间复杂度 O(N)