121. 买卖股票的最佳时机
- 买卖股票的最佳时机
思路
确定dp数组(dp table)以及下标的含义
dp[i][0] 表示第i天持有股票所得最多现金 ,这里可能有同学疑惑,本题中只能买卖一次,持有股票之后哪还有现金呢?
其实一开始现金是0,那么加入第i天买入股票现金就是 -prices[i], 这是一个负数。
dp[i][1] 表示第i天不持有股票所得最多现金
注意这里说的是“持有”,“持有”不代表就是当天“买入”!也有可能是昨天就买入了,今天保持持有的状态
思路代码
func maxProfit(prices []int) int {
dp:=make([]int,2)
dp[0]=-prices[0]
dp[1]=0
for i:=1;i<len(prices);i++{
dp[0],dp[1]=max(dp[0],-prices[i]),max(prices[i]+dp[0],dp[1])
}
return dp[1]
}
func max(i,j int)int{
if i<j{
return j
}
return i
}
困难
dp[0],dp[1]=max(dp[0],-prices[i]),max(prices[i]+dp[0],dp[1])
122.买卖股票的最佳时机II
思路
如果第i天持有股票即dp[i][0], 那么可以由两个状态推出来
第i-1天就持有股票,那么就保持现状,所得现金就是昨天持有股票的所得现金 即:dp[i - 1][0]
第i天买入股票,所得现金就是昨天不持有股票的所得现金减去 今天的股票价格 即:dp[i - 1][1] - prices[i]
注意这里和121. 买卖股票的最佳时机 (opens new window)唯一不同的地方,就是推导dp[i][0]的时候,第i天买入股票的情况。
思路代码
func maxProfit(prices []int) int {
dp:=make([]int,2)
dp[0]=-prices[0]
dp[1]=0
for i:=0;i<len(prices);i++{
dp[0],dp[1]=max(dp[0],dp[1]-prices[i]),max(dp[1],dp[0]+prices[i])
}
return dp[1]
}
func max(i,j int)int{
if i<j{
return j
}else{
return i
}
}
困难
在121. 买卖股票的最佳时机 (opens new window)中,因为股票全程只能买卖一次,所以如果买入股票,那么第i天持有股票即dp[i][0]一定就是 -prices[i]。
而本题,因为一只股票可以买卖多次,所以当第i天买入股票的时候,所持有的现金可能有之前买卖过的利润。
那么第i天持有股票即dp[i][0],如果是第i天买入股票,所得现金就是昨天不持有股票的所得现金 减去 今天的股票价格 即:dp[i - 1][1] - prices[i]。
今日收获
股票问题动规解法
dp[0]代表持有股票
dp[1]代表不持有股票