给你一个整数数组 prices
和一个整数 k
,其中 prices[i]
是某支给定的股票在第 i
天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k
笔交易。也就是说,你最多可以买 k
次,卖 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 。
提示:
1 <= k <= 100
1 <= prices.length <= 1000
0 <= prices[i] <= 1000
题解:
/**
* @description: 动态规划 TC:O(n^2) SC:O(n)
* @param {*} k 给定一个整数,表示买卖次数
* @param {*} prices 给定一个数组,表示股票趋势
*/
function dp(k,prices){
/**
* 本方案使用DP,定义两数组buy、profit,buy中
* 元素表示,第n次买入后的剩余利润,buy[0]除
* 外,buy[0]表示第1次买入的成本; profit中
* 元素表示第n次卖出后的利润。
*
* 其中
*
* buy[0]=min(prices[i],buy[0])
* profit[0]=max(prices[i]-buy[0], profit[0])
*
* buy[n]=max(profit[n-1]-prices[i],buy[n])
* profit[n]=max(buy[n]+prices[i],profit[n])
*
*
* 这是买卖股票的模板,无论1次、2次还是k次
* 都可以使用这份模板实现。
*/
let buy=new Array(k).fill(-prices[0]),
profit=new Array(k).fill(0);
buy[0]=prices[0]
for(let i=1;i<prices.length;i++){
buy[0]=Math.min(buy[0],prices[i])
profit[0]=Math.max(profit[0],prices[i]-buy[0])
for(let j=1;j<k;j++){
buy[j]=Math.max(buy[j],profit[j-1]-prices[i])
profit[j]=Math.max(profit[j],buy[j]+prices[i])
}
}
return profit[k-1];
}