题目1: 121. 买卖股票的最佳时机
算法思路:
最多总共只能买卖一次。
dp[i][0] 表示持有股票的最大收益。
dp[i][1]表示未持有股票的最大收益。
代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if(n == 0) return 0;
vector<vector<int>> dp(n, vector<int>(2));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for(int i = 1; i < n; 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]);
}
return dp[n - 1][1];
}
};
题目 2: 122. 买卖股票的最佳时机 II
算法思路:
跟上题类似,只是可以多次买卖,但最多只可以持有一支股票。
代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if(n == 0) return 0;
vector<vector<int>> dp(n, vector<int>(2));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for(int i = 1; i < n; i++) {
dp[i][0] = max(dp[i-1][0], dp[i-1][1]-prices[i]);
dp[i][1] = max(dp[i-1][1], dp[i-1][0]+prices[i]);
}
return dp[n-1][1];
}
};
题目3: 123. 买卖股票的最佳时机 III
算法思路:
跟上题类似,只是可以多次买卖,并且最多可以持有2支股票。
dp[i][0] 表示未执行任何操作
dp[i][1] 表示持有第一支股票的收益
dp[i][2]表示未持有第一支股票的收益
dp[i][3] 表示持有第二支股票的收益
dp[i][4]表示未持有第二只股票的收益
代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
if(n == 0) return 0;
vector<vector<int>> dp(n, vector<int>(5));
dp[0][1] = -prices[0];
dp[0][3] = -prices[0];
for(int i = 1; i < n; i++) {
dp[i][0] = dp[i - 1][0];
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]);
dp[i][4] = max(dp[i - 1][4], dp[i - 1][3] + prices[i]);
}
return dp[n - 1][4];
}
};