题目:
给定一个整数数组prices表示每一天的股票票价,最多可以进行k次交易,最大利润是多少。(你只能在卖出这支股票之后才能进行买入操作)
1. 可以定义两个状态数组buy和sell。
buy[i][j]表示在第i支股票时,进行了j次交易的最大利润,并且此时手上还有一支股票。
sell[i][j]表示正在第i支股票时,进行了j次交易的最大利润。
2. 状态转移方程:
buy[i][j] = max{buy[i-1][j], sell[i-1][j] - prices[i]}
可以是在第i天之前买入的,也可以是第i天买入的。
sell[i][j] = max{sell[i-1][j], buy[i-1][j-1] + prices[i]}
可以在第i天出售股票,也可以在第i天之前出售。
3. 初始化dp数组
buy[i][0] 初始化为-prices[0],表示在进行交易之前买入的股票。
sell[i][0]初始化为0,表示没有进行交易时的最大利润为0。
4. 先遍历股票,再遍历交易。
5. 优化,观察上面的状态转移方程可以看出支股票的状态都由上一支股票转移而来,因此可以开辟一个一维数组即可。
此时的状态转移方程如下:
buy[j] = max{buy[j], buy[j] - prices[i]}
sell[j] = max{sell[j], buy[j -1] + prices[i]}
func maxProfit(k int, prices []int) int {
n := len(prices)
if k == 0|| n == 0{
return 0
}
buy := make([]int, k+1)
sell := make([]int, k+1)
buy[0] = -prices[0]
sell[0] = 0
for i:=1; i<= k;i++{
buy[i] = math.MinInt32
sell[i] = math.MinInt32
}
ans := 0
for i:=1; i< n;i++{
buy[0] = max(buy[0], sell[0] - prices[i])
for j:=1;j <= k;j++{
buy[j] = max(buy[j], sell[j] - prices[i])
sell[j] = max(sell[j], buy[j-1] + prices[i])
}
}
for i:=0; i<=k;i++{
ans = max(ans, sell[i])
}
return ans
}
func max(a, b int)int{
if a > b{
return a
}
return b
}