[leetcode] 309. Best Time to Buy and Sell Stock with Cooldown

问题拆解:

对于股票,每天只有两种状态:手中有股票,手中无股票。
对于手中有股票进行拆解:
 1. 前一天手中有股票,今天什么都不做
 2. 前一天手中无股票,并且前一天没有卖股票,今天买股票
对于手中无股票进行拆解:
 1. 前一天手中无股票,今天什么都不做
 2. 前一天手中有股票,今天卖股票
因此可以设:
dpi0表示当天手中有股票的最大获利
dpi1表示当天手中无股票的最大获利
dpi2表示当天手中无股票且前一天未卖股票的最大获利(必须要有dpi2,因为dpi1是否能买股票取决于dpi2))
12.可得状态方程如下:
dpi0 = max(dpi0, dpi2 - prices[i]); // 此处减去prices[i]表示买股票的花费,后续卖股票的获利直接加上prices[i]就行,这样就无需在卖股票的时候关心在哪里买的
dpi1 = max(dpi1, dpi0 + prices[i]); // 注意实际编码过程中dpi0被覆盖了,状态转移的顺序需要调换
dpi2 = dpi1;
最后获利的最大值为max(dpi1, dpi2);
代码块
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        // dpi0表示第i天手中持有股票的最大获利
        // dpi1表示第i天手中无股票的最大获利
        // dpi2表示第i天手中无股票且不卖出股票的最大获利
        int n = prices.size();
        int dpi0 = INT_MIN, dpi1 = 0, dpi2 = 0;
        for (int i = 0; i < n; ++i) {
            int tmp = dpi0;
            dpi0 = max(dpi0, dpi2 - prices[i]);
            dpi2 = dpi1;
            dpi1 = max(dpi1, tmp + prices[i]);
        }
        
        return max(dpi1, dpi2);
    }
};

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if (!n)return 0;
        vector<int>buy(n), sell(n);
        buy[0] = -prices[0];
        sell[0] = 0;
        for (int i = 1; i < n; ++i) {
            sell[i] = max(sell[i - 1], buy[i - 1] + prices[i]);
            if (i < 2)
                buy[i] = max(buy[i - 1], -prices[i]);
            else buy[i] = max(buy[i - 1], sell[i - 2] - prices[i]);
        }
        return sell[n - 1];
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值