题目来源:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-iv/description/
C++题解:动态规划
思路同力扣 123. 买卖股票的最佳时机 III-CSDN博客,只是把最高2次换成k次。如果思路不清晰,可以将k从0写到4等找找规律。
class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
int n = prices.size();
if(n <= 1) return 0;
// dp[i][j], 当j为偶数表示第j/2+1次买入,奇数表示(j+1)/2次卖出
vector< vector<int> > dp(n, vector<int>(2*k, 0));
for(int j = 0; j < k; j++) {
dp[0][j*2] = -prices[0]; //初始化
}
for(int i = 1; i < n; i++) {
dp[i][0] = max(dp[i-1][0], -prices[i]);
for(int j = 1; j < k; j++){
dp[i][2*j] = max(dp[i-1][2*j], dp[i-1][2*j-1] - prices[i]);
}
for(int m = 0; m < k; m++) {
dp[i][2*m+1] = max(dp[i-1][2*m+1], dp[i-1][2*m] + prices[i]);
}
}
return dp[n-1][2*k-1];
}
};