【力扣】股票买卖系列问题的状态机解法

力扣股票系列题目汇总

贪心：

• 121：最多进行一次买卖
• 122：可以多次买卖

DP：

• 309：多次买卖+卖出冷冻期
• 714：多次买卖+卖出手续费
• 123：最多两次买卖
• 188：最多 k 次买卖

121 买卖股票的最佳时机

给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。

你只能选择 某一天 买入这只股票，并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。

返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0 。

思路：贪心

要顺序地找到数组最小的元素，然后找到最大的元素，差值即为最大利润

代码

//时间复杂度O(n) 空间复杂度O(1)
class Solution {
   
public:
    int maxProfit(vector<int>& prices) {
   
        int buy = INT_MAX, sell = INT_MIN;
        for (int& price : prices){
   
            // 记录历史最低价格
            buy = min(buy, price);
            // 记录历史最大收益
            sell = max(sell, price - buy);
        }
        return sell;
    }
};

122 买卖股票的最佳时机 II

给定一个数组 prices ，其中 prices[i] 表示股票第 i 天的价格。

在每一天，你可能会决定购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以购买它，然后在 同一天 出售。
返回 你能获得的 最大 利润 。

思路：贪心

只要比前一天价格高，就卖出；卖出的同时买入，因为第二天有个min函数决定是否真的买入

代码

//时间复杂度O(n) 空间复杂度O(1)
class Solution {
   
public:
    // 只要比前一天高，就卖出
    int maxProfit(vector<int>& prices) {
   
        int buy = INT_MAX, profit = 0;
        for (int i = 0; i < prices.size(); ++i){
   
            buy = min(buy, prices[i]);
            if (prices[i] > buy) {
   
                profit += prices[i] - buy;
                buy = prices[i];
            }
        }
        return profit;
    }
};

优化

发现buy和 prices[i]的大小比较进行了两次，所以将循环体内的min函数去掉。

//时间复杂度O(n) 空间复杂度O(1)
class Solution {
   
public:
    int maxProfit(vector<int>& prices) {
   
        int profit = 0;
        for (int i = 1; i < prices.size(); ++i){
     
            if (prices[i] > prices[i-1]) {
   
                pr