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

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

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
示例:

输入: [1,2,3,0,2]
输出: 3 
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

基本思路：动态规划

• 定义：dp[i][j],表示的是第i天的状态为j,其中，0表示的是不是因为今天卖出了股票的不持有状态，1表示的是持股，可以卖，2表示的是由于今天卖出的不持有状态
• 状态转移方程如下：

 

 

    int maxProfit(vector<int>& prices) {
        if(prices.size()<1)
            return 0;
        int len=prices.size();
        vector<vector<int>> dp(prices.size()+1,vector<int>(3,0));
        dp[0][0]=0; //手中没有持股，可以买
        dp[0][1]=-prices[0]; //手中有股，今天刚买，可以卖
        dp[0][2]=0;  //手中未持有，今天刚卖出，处于冷冻期，明天不能买入。
        for(int i=1;i<prices.size();i++){  //状态转移方程
            dp[i][0]=max(dp[i-1][0],dp[i-1][2]);
            dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1]);
            dp[i][2]=dp[i-1][1]+prices[i];
        }

        return max(dp[len-1][0],dp[len-1][2]);//最后结果一定是处于未持有状态，因为若持有，则定小于dp[0],亏损状态。
    }

基本思路：空间压缩，只与前一个状态有关。

    int maxProfit(vector<int>& prices) {
        if(prices.size()<1)
            return 0;
        int len=prices.size();
        vector<int> dp(3,0);
        dp[0]=0; //手中没有持股，可以买
        dp[1]=-prices[0]; //手中有股，今天刚买
        dp[2]=0;  //手中未持有，今天刚卖出，处于冷冻期，明天不能买入。
        for(int i=1;i<prices.size();i++){
            int dp0=max(dp[0],dp[2]);
            int dp1=max(dp[0]-prices[i],dp[1]);
            dp[2]=dp[1]+prices[i];
            dp[0]=dp0;
            dp[1]=dp1;
        }

        return max(dp[0],dp[2]);//最后结果一定是处于未持有状态，因为若持有，则定小于dp[0],亏损状态。
    }