力扣刷题 122.买卖股票的最佳时机II

题干

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

在每一天，你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买，然后在 同一天 出售。

返回 你能获得的 最大 利润 。

示例 1：

输入：prices = [7,1,5,3,6,4]
输出：7
解释：在第 2 天（股票价格 = 1）的时候买入，在第 3 天（股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4 。
     随后，在第 4 天（股票价格 = 3）的时候买入，在第 5 天（股票价格 = 6）的时候卖出, 这笔交易所能获得利润 = 6 - 3 = 3 。
     总利润为 4 + 3 = 7 。

示例 2：

输入：prices = [1,2,3,4,5]
输出：4
解释：在第 1 天（股票价格 = 1）的时候买入，在第 5 天 （股票价格 = 5）的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4 。
     总利润为 4 。

示例 3：

输入：prices = [7,6,4,3,1]
输出：0
解释：在这种情况下, 交易无法获得正利润，所以不参与交易可以获得最大利润，最大利润为 0 。

---

解题思路

这道题目我们用到的思想还是贪心，经过观察后，我们发现当股票上涨时，在最高点抛出可以得到最大的利润，换句话说就是，一旦股票下跌，我们就在下跌前卖出。因此局部最优就是在每次下跌前卖出，并重新买入。

这是笔者一开始写的完整代码

其中笔者设置了buy和sell两个变量来记录买入值和卖出值。在写代码中有不少细节需要注意，比较麻烦。比如说股票可能到最后也没有下跌的情况、刚刚买入股票还不知道第二天的价格，无法更新利润值。还有数组越界的情况。因此我们换另一种解法。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int buy =prices[0];//买入初始值为第一天价格
        int sell = 0;//卖出值初始值为0
        int pro = 0;//pro记录利润
        for(int i =0; i<prices.size();i++){
        if(i == prices.size()-1||prices[i] > prices[i+1]){
            if(sell != 0){//当已经有比买入值更高的价格时
            pro += (sell - buy);
            }
            if(i != prices.size()-1)//避免数组越界，比如数组为[1]
            buy = prices[i+1];//重新买入
            sell = 0;//卖出值设为0，因为还不知道第二天的价格
        }
        else if(prices[i] <= prices[i+1]){
                sell =prices[i+1];//当价格上涨，卖出值更新
            }
        
        }
    return pro;
    }
};

另一种解法的思路还是贪心，只不过有一个很重要且一时想不到的点是，我们可以把每天的利润拆开看，我们之前的思路是找到递增区间首尾相减算利润，但事实上，递增区间的利润等于每天由于价格上涨带来的利润之和，比如prices[4]-prices[1]=(prices[4]-prices[3])+(prices[3]-prices[2])+(prices[2] - prices[1]) + (prices[1] - prices[0])。此时就是把利润分解为每天为单位的维度，而不是从 2 天到第 5天整体去考虑。

因此在这个基础上我们的局部最优就变成了收集每天的正利润，而全局最优是求得最大利润。

以下是完整代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int profit = 0;
        for(int i = 0; i < prices.size() -1; i++){
            int diff = prices[i+1] - prices[i];
            if(diff > 0){
                profit += diff;
            }
        }
        return profit;
    }
};

代码中有一个细节需要注意，i不能遍历到最后一个元素，不然在算差值求利润的时候就会导致数组越界。因为prices[i+1]可能越界。