代码随想录算法训练营 day49|121. 买卖股票的最佳时机、122.买卖股票的最佳时机II

一、121. 买卖股票的最佳时机

力扣题目链接

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], prices[i] + dp[i - 1][0]);
        }
        return dp[len - 1][1];
    }
};

二、122.买卖股票的最佳时机II

力扣题目链接

与第一题的唯一区别就在于转移方程中dp[i][0]的状态变化，也就是第i天持有股票的最大收益，因为第一题只能买一次，所以如果前一天不持有今天持有的话一定是-prices[i]，但是这道题可以多次操作，所以前一天不持有今天持有的话应该是dp[i-1][1]-prices[i]。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        vector<vector<int>> dp(len, vector<int>(2, 0));
        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]); // 注意这里是和121. 买卖股票的最佳时机唯一不同的地方。
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
        }
        return dp[len - 1][1];
    }
};