Day42 动态规划 — 买卖股票

121. 买卖股票的最佳时机

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.size() == 0) return 0;
        
        vector<vector<int>> dp(prices.size(), vector<int>(2)); // 0 = 持有 ， 1 = 不持有
        dp[0][0] -= prices[0];
        dp[0][1] = 0;

        for (int i = 1; i < prices.size(); i++) {
            dp[i][0] = std::max(dp[i-1][0], -prices[i]); // 昨天持有、今天购入中的最大值
            dp[i][1] = std::max(dp[i-1][1], prices[i] + dp[i-1][0]); // 昨天持有、今天卖出的最大值
        }

        return dp[prices.size() - 1][1];
    }
};

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

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.size() == 0) return 0;

        vector<vector<int>> dp(prices.size(), vector<int>(2, 0)); // 0 = 持有， 1 = 不持有
        dp[0][0] = -prices[0];
        dp[0][1] = 0;

        for (int i = 1; i < prices.size(); i++) {
            dp[i][0] = std::max(dp[i-1][0], dp[i-1][1] - prices[i]);
            dp[i][1] = std::max(dp[i-1][1], dp[i-1][0] + prices[i]);
        }

        return dp[prices.size()-1][1];
    }
};