买卖股票的最佳时机IV
TIP
给定一个整数数组 prices
,它的第i
个元素 prices[i]
是一支给定的股票在第 i
天的价格,和一个整型 k
。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k
笔交易。也就是说,你最多可以买 k
次,卖 k
次。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
思路
- 第一次持有股票
- 第一次不持有股票
- 第二次持有股票
- 第二次不持有股票
- ……
- 第k次持有股票
- 第k次不持有股票
js
var maxProfit = function(k, prices) {
let dp = Array(prices.length).fill().map(() => Array(2*k+1).fill(0))
for(let i=1;i<2*k;i+=2){ //注意初始化!!! 只要买入,现金就做相应的减少
dp[0][i] -= prices[0]
}
for(let i=1;i<prices.length;i++){
for (let j = 0; j <=2*k - 1; j += 2) {
dp[i][j + 1] = Math.max(dp[i-1][j+1], dp[i-1][j] - prices[i]);
dp[i][j + 2] = Math.max(dp[i-1][j+2], dp[i-1][j+1] + prices[i]);
}
}
console.log(dp)
return dp[prices.length-1][2*k]
};