leecode 309. 最佳买卖股票时机含冷冻期

/*****
//sell[i]表示截至第i天，最后一个操作是卖时的最大收益；
//buy[i]表示截至第i天，最后一个操作是买时的最大收益；
//cool[i]表示截至第i天，最后一个操作是冷冻期时的最大收益；
//递推公式：
//sell[i] = max(buy[i-1]+prices[i], sell[i-1]) (第一项表示第i天卖出，第二项表示第i天冷冻)
//buy[i] = max(cool[i-1]-prices[i], buy[i-1]) （第一项表示第i天买进，第二项表示第i天冷冻）
//cool[i] = max(sell[i-1], cool[i-1]) (第一项表示第i-1天卖出，第二项表示第i-1天之前卖出）
****/
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len=prices.size();
        if(len<=1) return 0;
        vector<int>sell(len,0);
        vector<int>buy(len,-prices[0]);
        vector<int>cool(len,0);
        for(int i=1;i<len;i++){
            int p=prices[i];
            sell[i]=max(buy[i-1]+p,sell[i-1]);//
            buy[i]=max(buy[i-1],cool[i-1]-p);//前一天必须冷冻期才能买入
            cool[i]=max(sell[i-1],cool[i-1]);//前一天要么卖出，要么是冷冻期
        }
        return sell[len-1];
    }
};