买卖股票的最佳时机(交易次数限制)
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
状态转移方程
buy[i][j]=max{buy[i−1][j],sell[i−1][j]−price[I]}
sell[i][j]=max{sell[i−1][j],buy[i−1][j−1]+price[i]}
需要注意卖出股票操作时,需要从购买次数 - 1
的状态转移。
确定初始状态也很重要,buy[0][0]
代表第一天便买入的状态,而buy[i][0]
表示1~n-1天每天卖出次数为0的状态,次状态只需要为买入前i天的最小价格的股票即可。
而buy[0][j]
为第一天买卖次数为1 ~ k的状态,不能从此状态转移,所以全部赋为0。
最后结果为最后一天,对所有卖出次数求最大值。
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size(),k = 2;
//表示n天进行了k次交易
vector<vector<int>> buy(n, vector<int>(k + 1));
vector<vector<int>> sell(n, vector<int>(k + 1));
buy[0][0] = -prices[0];
sell[0][0] = 0;
for (int i = 1; i <= k; ++i) {
buy[0][i] = sell[0][i] = INT_MIN / 2;
}
for (int i = 1; i < n; ++i) {
buy[i][0] = max(buy[i - 1][0], sell[i - 1][0] - prices[i]);
for (int j = 1; j <= k; ++j) {
buy[i][j] = max(buy[i - 1][j],sell[i - 1][j] - prices[i]);
sell[i][j] = max(buy[i - 1][j - 1] + prices[i],sell[i - 1][j]);
}
}
return *max_element(sell[n - 1].begin(), sell[n - 1].end());
}
};