题目:
给定一个数组 prices ,其中 prices[i] 是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
思路:
假设最开始的时候有0元,如果能够知道第i天收盘手上最多的钱,那么就能够知道第i天收盘最大的利润。设置一个二维dp数组。dp[i][0]表示第i天收盘 手上没有股票时,手上最多的金额,dp[i][1]表示第i天收盘 手上有股票时,手上最多的金额。dp[0][0]=0,dp[0][1]=-prices[0]
容易第i天手上最多的金额数和i-1天最多的金额数有以下对应关系:
dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i])
dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1])
class Solution:
def maxProfit(self, prices):
dp=[[1,1] for i in range(len(prices))]
dp[0][0]=0
dp[0][1]=-prices[0]
for i in range(1,len(prices)):
dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i])
dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1])
return dp[-1][0]
时间复杂度:O(N)
空间复杂度:O(N)