121. 买卖股票的最佳时机
买卖股票和打家劫舍一样,都有两种状态,买还是不买,卖还是不卖,因此dp[i][0]表示第i天持有股票的最大现金,dp[i][1]表示第i天不持有股票的现金。
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(), vector<int>(2, 0));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for (int i = 1; i < dp.size(); ++i) {
dp[i][0] = max(dp[i - 1][0], -prices[i]);
dp[i][1] = max(dp[i - 1][1], prices[i] + dp[i - 1][0]);
}
return dp.back()[1];
}
};
122.买卖股票的最佳时机II
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(), vector<int>(2, 0));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for (int i = 1; i < dp.size(); ++i) {
dp[i][0] = max(dp[i - 1][0], dp[i-1][1] - prices[i]);
dp[i][1] = max(dp[i - 1][1], prices[i] + dp[i - 1][0]);
}
return dp.back()[1];
}
};
123.买卖股票的最佳时机III
class Solution {
public:
int maxProfit(vector<int>& prices) {
// dp[i][0]:第一次持有
// dp[i][1]:第一次不持有
// dp[i][2]:第二次持有
// dp[i][3]:第三次持有
// 递推公式和前面差不多
// 初始化需要考虑
vector<vector<int>> dp(prices.size(), vector<int>(4, 0));
dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[0][2] = -prices[0];
dp[0][3] = 0;
for (int i = 1; i < prices.size(); ++i) {
dp[i][0] = max(dp[i - 1][0], -prices[i]);
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
dp[i][2] = max(dp[i - 1][2], dp[i - 1][1] - prices[i]);
dp[i][3] = max(dp[i - 1][3], dp[i - 1][2] + prices[i]);
}
return dp.back()[3];
}
};