给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入:k = 2, prices = [2,4,1]
输出:2
解释:在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。
示例 2:
输入:k = 2, prices = [3,2,6,5,0,3]
输出:7
解释:在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。
提示:
0 <= k <= 100
0 <= prices.length <= 1000
0 <= prices[i] <= 1000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
- 思路看代码注释
这道题是(123.买卖股票的最佳时机 III) 的升级版,区别是123题是固定交易2次,而这道题是的交易次数是变量K,搞懂这个题第123题自然也就解决了 - 代码
package main
import "fmt"
func maxProfit(k int, prices []int) int {
if len(prices) == 0 {
return 0
}
var dp = make([][101][2]int,len(prices))
for i:=0; i<=k; i++ {
dp[0][i][0] = 0
dp[0][i][1] = -prices[0]
}
for i:=1; i<len(prices); i++ {
for ki:=1; ki<=k; ki++ {
dp[i][ki][0] = max(dp[i-1][ki][0],max(dp[i-1][ki-1][0],dp[i-1][ki][1]+prices[i]))
dp[i][ki][1] = max(dp[i-1][ki][1],dp[i-1][ki-1][0]-prices[i])
}
}
return dp[len(prices)-1][k][0]
}
func max(a int, b int) int {
if a > b {
return a
}
return b
}
func main() {
nums := []int{3,2,6,5,0,3}
res := maxProfit(2,nums)
fmt.Println(res)
}