复杂动态规划之状态机解法

刷到leetcode309 最佳买卖股票时机含冷冻期时,学习了一种状态机的解法,觉得很好用,遂记录一下。

在这里插入图片描述
对于这类复杂的状态转移问题,可以通过建立多个状态以及他们之间的转移方程来解决。对于该题,则定义四种状态,买入,卖出,持有,和不持有,其值代表当前状态下最大的利润。

状态机及之间的转换如下:
在这里插入图片描述

#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n=prices.size();
        if(n==0)
            return 0;
        //定义四种状态,买入,卖出,持有,和不持有,其值代表当前状态下最大的利润
        vector<int> buy(n), sell(n), hold(n), not_hold(n);
        //初始化
        hold[0]=buy[0]=-prices[0];  //意思是上来就买入/持有股票,的最大收益
        sell[0]=not_hold[0]=0; //意思是上来就卖出/不持有股票,的最大收益
        for(int i=1;i<n;i++){
            //只能在冻结期后进行买入
            buy[i]=not_hold[i-1]-prices[i];
            //可以在买入后和已经持有的情况下,继续持有
            hold[i]=max(buy[i-1],hold[i-1]);
            //只能在买入后和已经持有的情况下,卖出
            sell[i]=max(buy[i-1],hold[i-1])+prices[i];
            //只能在卖出后或者不持有后,保持不持有
            not_hold[i]=max(not_hold[i-1],sell[i-1]);
        }
        //只有卖出和非持有状态,才可能有最高的利润
        return max(sell[n-1],not_hold[n-1]);
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值