- 题目要求进行k次买卖
- 其实就是上一题的扩展,把2次扩展为k次
- 定义动规数组依然是二维,第一个维度表示第几天,第二个维度表示第几次买入和卖出
- 所以第二个维度的长度应该是
2k+1
- 在for循环内,要使用一个内循环来表示第几次买入或卖出,根据上一题的过程,可以发现其中的规律
class Solution {
public:
int maxProfit(int k, std::vector<int>& prices) {
int len = prices.size();
if (len == 0) {
return 0;
}
std::vector<std::vector<int>> dp(len, std::vector<int>(2 * k + 2, 0));
for (int i = 0; i < 2 * k; i += 2) {
dp[0][i + 1] = -prices[0];
}
for (int i = 1; i < len; ++i) {
for (int j = 0; j < 2 * k; j += 2) {
dp[i][j + 1] = std::max(dp[i - 1][j + 1], dp[i - 1][j] - prices[i]);
dp[i][j + 2] = std::max(dp[i - 1][j + 2], dp[i - 1][j + 1] + prices[i]);
}
}
return dp[len - 1][2 * k];
}
};