Leetcode #309 Best Time to Buy and Sell Stock with Cooldown

Best Time to Buy and Sell Stock with Cooldown

题目描述
有一只股票，股票的每天的价格用一个整数序列给出。
操作要求如下：

• 当你持有股票的时候，你可以卖出。卖出之后，你下一天不能进行任何买卖行为(cooldown)。
• 当你不持有任何股票的时候，你可以买进。

问最后能获得的最大利润是多少。

Example：
prices = [1, 2, 3, 0, 2]
maxProfit = 3
transactions = [buy, sell, cooldown, buy, sell]

算法分析
令$f_0[i]$表示第i天买进股票所能获得的最大利润。
令$f_1[i]$表示第i天卖出股票所能获得的最大利润。
那么我们可以得出下述两个状态转移方程：

• $f_0[i]=max(f_1[j])-prices[i], j<i-1$
• $f_1[i]=max(f_0[j])+prices[i], j<i$

然后，我们需要用两个变量来维护 $max(f_1[j]), j<i-1$ 和 $max(f_0[j]), j<i$
最后，最大利润即为$max(f_1[i])，1<=i<=n$
算法时间复杂度：O(n)

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n=prices.size();
        if (n<2) return 0;
        vector<int> f0(n);
        vector<int> f1(n);
        f0[0]=-prices[0];
        f1[0]=0;
        f0[1]=-prices[1];
        f1[1]=max(prices[1]-prices[0],0);
        int m1=f1[0];
        int m0=max(f0[0],f0[1]);
        int ans=f1[1];
        for (int i=2;i<n;i++)
        {
            f0[i]=m1-prices[i];
            f1[i]=m0+prices[i];
            if (f1[i]>ans) ans=f1[i];
            m0=max(m0,f0[i]);
            m1=max(m1,f1[i-1]); 
        }
        return ans;
    }
};