best-time-to-buy-and-sell-stock-with-cooldown.cpp

// Time:  O(n)
// Space: O(1)

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.empty()) {
            return 0;
        }
        vector<int> buy(2), sell(2), coolDown(2);
        buy[0] = -prices[0];
        for (int i = 1; i < prices.size(); ++i) {
            // Bought before or buy today.
            buy[i % 2] = max(buy[(i - 1) % 2], coolDown[(i - 1) % 2] - prices[i]);
            // Sell today.
            sell[i % 2] = buy[(i - 1) % 2] + prices[i];
            // Sold before yesterday or sold yesterday.
            coolDown[i % 2] = max(coolDown[(i - 1) % 2], sell[(i - 1) % 2]);
        }
        return max(coolDown[(prices.size() - 1) % 2], sell[(prices.size() - 1) % 2]);
    }
};

// Time:  O(n)
// Space: O(n)
class Solution2 {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.empty()) {
            return 0;
        }
        vector<int> buy(prices.size()), sell(prices.size()), coolDown(prices.size());
        buy[0] = -prices[0];
        for (int i = 1; i < prices.size(); ++i) {
            // Bought before or buy today.
            buy[i] = max(buy[i - 1], coolDown[i - 1] - prices[i]);
            // Sell today.
            sell[i] = buy[i - 1] + prices[i];
            // Sold before yesterday or sold yesterday.
            coolDown[i] = max(coolDown[i - 1], sell[i - 1]);
        }
        return max(coolDown[prices.size() - 1], sell[prices.size() - 1]);
    }
};