给定一个整数数组 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 <= 109
- 0 <= prices.length <= 1000
- 0 <= prices[i] <= 1000
class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
}
};
思路:
原来一看这题贼高兴,因为瞥一眼很熟悉,只不过这句只能完成 k 笔交易,表明了有回退的可能性,那就把每天可能的情况用数组记录下数值。
但后来还是没想明白只能完成 k 笔交易应该怎么实现,所以,瞄了很多眼题解进行学习。要格外注意边界问题。
class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
int len=prices.size();
if(len==0)
return 0;
k=min(k,len/2); //最大利润不一定要完成k笔交易
vector<vector<int>> vk0(len,vector<int>(k+1)); //当前手上无股票
vector<vector<int>> vk1(len,vector<int>(k+1)); //当前手上有股票
for(int j=1;j<=k;++j){ //因为第 0 天我们不会已经完成多次交易,设置一个相对较小的非法值
vk0[0][j]=INT_MIN/2; //因为后续会有vk0[i-1][j]-prices[i],直接设置最小会溢出
vk1[0][j]=INT_MIN/2;
}
vk0[0][0]=0;
vk1[0][0]=-prices[0];
for(int i=1;i<len;++i){
vk1[i][0]=max(vk1[i-1][0],vk0[i-1][0]-prices[i]); //假设从第i天才开始购入
for(int j=1;j<=k;++j){
vk0[i][j]=max(vk0[i-1][j],vk1[i-1][j-1]+prices[i]); //max(昨天无股票,昨天有股票今天卖了,以卖出作为交易次数加1)
vk1[i][j]=max(vk1[i-1][j],vk0[i-1][j]-prices[i]); //max(昨天有股票,昨天无股票今天买了)
}
}
int vmax=0;
for(int j=0;j<=k;++j)
vmax=max(vmax,vk0[len-1][j]);
return vmax;
}
};