LeetCode第 309 题:最佳买卖股票时机含冷冻期(C++)

309. 最佳买卖股票时机含冷冻期 - 力扣(LeetCode)
在这里插入图片描述
股票系列:
LeetCode第 121 题:买股票的最佳时机(C++)_zj-CSDN博客
LeetCode第 122 题:买股票的最佳时机II(C++)_zj-CSDN博客
LeetCode第 123 题:买股票的最佳时机III(C++)_zj-CSDN博客
LeetCode第 188 题:买股票的最佳时机IV(C++)_zj-CSDN博客
LeetCode第 309 题:最佳买卖股票时机含冷冻期(C++)_zj-CSDN博客
LeetCode第 714 题:买卖股票的最佳时机含手续费(C++)_zj-CSDN博客

相比之前的题目加了一个限制,交易次数依然是无限次。
LeetCode第 122 题:买股票的最佳时机II(C++)_zj-CSDN博客

那么只需要在之前的状态转移上做一点修改,也就是买入的时候,状态需要从前两天转移过来,其他的都是一样的:

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n < 2) return 0;
        //第一维表示日期,第二维表示是否持有股票(1:有,0:无),数组值表示收益
        vector<vector<int>> dp(n, vector<int>(2, -1e7));
        dp[0][0] = 0; //第一天没有股票,没买没卖,收益为0
        dp[0][1] = 0 - prices[0]; //第一天有股票,说明买入了,收益为负(花掉一笔钱)
        dp[1][0] = max(dp[0][0], dp[0][1]+prices[1]);//昨天就没有今天rest,或者昨天有今天进行了卖出
        dp[1][1] = max(dp[0][1], dp[0][0]-prices[1]);//昨天就有今天rest,或者昨天没有今天进行买入
        for(int i = 2; i < n; ++i){
            dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i]);//前一天就没有今天rest,或者前一天有今天卖出
            dp[i][1] = max(dp[i-1][1], dp[i-2][0] - prices[i]);//前一天就有今天rest,或者前两天卖出、今天买入
        }
        //return max(dp[n-1][0], dp[n-1][1]);要想收益最大,最后一天肯定不会进行买入,所以dp[n-1][0]肯定更大
        return dp[n-1][0];
    }
};

可以进行状态压缩,上述dp只涉及三个变量:

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n < 2) return 0;
        int a = 0, b = -prices[0];//第一天两种状态(未持有、持有)
        int c = max(a, b+prices[1]);//第二天未持有
        int d = max(b, a-prices[1]);//第二天持有
        for(int i = 2; i < n; ++i){
            int tmp1 = c;
            c = max(c, d+prices[i]);
            d = max(d, a-prices[i]);
            a = tmp1; 
        }
       return c;
    }
};

看了题解之后,发现其实状态的定义也可以再增加一个,也就是持有、未持有、冷冻期,不过总体思路都是差不多的。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值