309. Best Time to Buy and Sell Stock with Cooldown

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size(), i;
        if(n == 0)
        return 0;
        vector<int> buy(n, 0);
        vector<int> sell(n, 0);
        vector<int> cooldown(n, 0);
        buy[0] = -prices[0];
        for(i = 1; i < n; ++i)
        {
            buy[i] = max(cooldown[i-1]-prices[i], buy[i-1]);
            sell[i] = max(buy[i-1] + prices[i], sell[i-1]);
            cooldown[i] = sell[i-1];
        }
        return sell[n-1];
    }
};