Leetcode 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

超时的dfs:

    int ans;
    void dfs(vector<int> &prices,int idx,int pre,int sum){
        if(idx >= prices.size()){
            ans = max(ans,sum);
            return;
        }
        if(pre < 0){
            //buy
            dfs(prices,idx+1,prices[idx],sum);
            //no-buy
            dfs(prices,idx+1,pre,sum);
        }else{
            //sell
            dfs(prices,idx+2,-1,sum+prices[idx]-pre);
            //no-sell
            dfs(prices,idx+1,pre,sum);
        }
    }

 

动态规划

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