leetcode 188. 买卖股票的最佳时机 IV hard
题目描述:
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
解题思路:
跟上一题 leetcode123. 买卖股票的最佳时机3 解法是一样的
但是要注意一个tips:
当k大于数组的一半size的时候,转为无限制的买卖 也就是leetcode 122题
为什么是一半呢? 画一下k线图就理解了,n个交易日,最多只有 n/2 个 波峰和波谷
而最大利润 最多也就是把所有波行情都给吃了,并且每天吃和 一次吃完得到的钱是一样的
代码:
class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
if(k<1 || prices.size()<2)
return 0;
// 当k大于数组的一半size的时候,转为无限制的买卖 也就是leetcode 122题
// 为什么是一半呢? 画一下k线图就理解了,n个交易日,最多只有 n/2 个 波峰和波谷
// 而最大利润 最多也就是把所有波行情都给吃了,并且每天吃和 一次吃完得到的钱是一样的
if(k>prices.size()/2)
return nolimts(prices);
// 虽然是二维数组,但同样利用了空间压缩技巧,因为实际应该是三维数组
// 并且为了好写所以是k+1 (为了边界好写)
vector<vector<int>> dp(k+1,vector<int> (2));
//初值的设定,注意下下个for循环里面dp【0】【0】和dp【0】【1】 是一直不变的
for(int i=0;i<=k;++i){
dp[i][0]=0;
dp[i][1]=INT_MIN;
}
// 更新dp
for(int &price:prices){
//为了不覆盖,从k到1,第二个维度先0再1
for(int i=k;i>=1;--i){
dp[i][0]=max(dp[i][0],dp[i][1]+price);
dp[i][1]=max(dp[i][1],dp[i-1][0]-price);
}
}
return dp[k][0];
}
int nolimts(vector<int> &vec){
int res=0;
for(int i=1;i<vec.size();++i){
if(vec[i]>vec[i-1])
res += vec[i]-vec[i-1];
}
return res;
}
};