题目描述
给你一个整数数组 prices 和一个整数 k ,其中 prices[i] 是某支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。也就是说,你最多可以买 k 次,卖 k 次。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
代码
/*
dp含义
dp[i][0] 不操作
dp[i][1] 第一次持有:分两种情况,今天之前已经买入或今天买入
dp[i][2] 第一次卖出:分两种情况,今天之前已经卖出或今天卖出
dp[i][3] 第二次买入:分两种情况,今天之前已经买入或今天买入
dp[i][4] 第二次卖出:分两种情况,今天之前已经卖出或今天卖出
...
dp[i][2*k-1]:第k次持有
dp[i][2*k]:第k次卖出
递推公式:
dp[i][j] = max(dp[i-1][j],dp[i-1][j-1]-price[i]);
dp[i][j+1] = max(dp[i-1][j+1],dp[i-1][j]+priece[i]);
初始化:
for(int i=1;i<=2k;i+=2)
dp[0][i]=-price[0];
遍历顺序
for(int i=1;i<price.size();i++)
for(int j=1;j<=2*k;j+=2)
*/
class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
if (prices.size() <= 1) return 0;
vector<vector<int>> dp(prices.size(), vector<int>(2*k+1, 0));
for (int i = 1; i <= 2*k; i+=2) {
dp[0][i] = -prices[0];
}
for (int i = 1; i < prices.size(); i++) {
for (int j = 1; j <= 2*k; j+=2) {
dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - 1] - prices[i]); //买入
dp[i][j + 1] = max(dp[i - 1][j + 1], dp[i - 1][j] + prices[i]); //卖出
}
}
return dp[prices.size() - 1][2*k];
}
};