Leetcode 309 买卖股票(含冷冻期) C++实现

Leetcode 309. 买卖股票的最佳时机（含冷冻期）

问题：给定一个整数数组 prices，其中第  prices[i] 表示第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票），卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

算法1：递归搜索 + 保存计算结果 = 记忆化搜索

        在Leetcode122. 买卖股票的基础上，只需修改一处：在计算持有股票的状态时，把 dfs ( i − 1 , 0 ) 改成 dfs ( i − 2 , 0) 。道理和Leetcode 198. 打家劫舍是一样的，因为第 i 天买股票的话第 i − 1 天不能卖，只能从第 i−2 天没有股票的状态转移过来。注意 dfs ( i − 2 , 0 ) 并不意味着第 i − 2 天一定卖了股票，而是在没有股票下的最优状态。

        请注意，这会导致边界条件多了一个 dfs ( − 2 , 0 ) = 0，后面空间优化中的 pre0​ 指的就是这个状态。

        请注意，dfs ( −  2 , 1 ) 是访问不到的，所以下面翻译成递推时，无需初始化这个状态（不需要写 dp [ 0  ] [ 1 ] = − ∞ ）。

代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        vector<array<int,2>> memo(n,{-1,-1});
        auto dfs = [&](auto &&dfs,int i,bool hold) -> int{
            if(i < 0)   return hold ? INT_MIN : 0;
            int& res = memo[i][hold];
            if(res != -1)   return res;
            if(hold)    return res = max(dfs(dfs,i - 1,true),dfs(dfs,i - 2,false) - prices[i]);
            return res = max(dfs(dfs,i - 1,false),dfs(dfs,i - 1,true) + prices[i]);
        };
        return dfs(dfs,n - 1,false);
    }
};

算法2：1:1 翻译成递推

代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        vector<array<int, 2>> dp(n + 2);
        dp[1][1] = INT_MIN;
        for(int i = 0;i < n;i++){
            dp[i + 2][0] = max(dp[i + 1][0],dp[i + 1][1] + prices[i]);
            dp[i + 2][1] = max(dp[i + 1][1],dp[i][0] - prices[i]);
        }
        return dp[n + 1][0];
    }
};

算法3：空间优化

代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int pre0 = 0,dp0 = 0,dp1 = INT_MIN;
        for(int p : prices){
            int new_dp0 = max(dp0,dp1 + p);
            dp1 = max(dp1,pre0 - p); // dp1不可以直接和dp0比，因为必须隔一天才能买
            pre0 = dp0;
            dp0 = new_dp0;
        }
        return dp0;
    }
};