第一题、买卖股票的最佳时机 力扣题目链接
贪心法:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int low = INT_MAX;
int result = 0;
for(int i=0; i < prices.size(); i++){
low = min(low, prices[i]);
result = max(result, prices[i] - low);
}
return result;
}
};
动态规划:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int len = prices.size();
if(len == 0) return 0;
vector<vector<int>> dp(len, vector<int>(2));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for(int i = 1; i < len; 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[len-1][1]; // 不持有的时候一定比持有的时候收益高
}
};
第二题、买卖股票的最佳时机II 力扣题目链接
0表示持有,1表示不持有。
今日持有的收益=比较昨天持有和昨天不持有今天持有收益谁更大;
今日不持有的收益=比较昨天不持有和昨天持有今天不持有的收益谁更大。
最后返回最终不持有的收益
class Solution {
public:
int maxProfit(vector<int>& prices) {
int len = prices.size();
vector<vector<int>> dp(len, vector<int>(2));
dp[0][0] = -prices[0]; // 持有股票的收益
dp[0][1] = 0; // 不持有的收益
for(int i=1; i < len; 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[len-1][1];
}
};