买股票的最佳时机含冷冻期--(动态规划(简单多状态dp问题))

题目

在这里插入图片描述

思路

状态表示

根据题目可以看出是从0天开始到后面,顺序是从左往右,同样根据经验+题目要求,经验是以i位置为终点(或起点)。
这个题目:dp[i]表示以i位置为终点,获得的最大利润。
在第i天结束后有三种状态,比较复杂,所以需要用二维dp表来表示,dp[i][0]表示在第i天结束后处于“买入”状态的最大利润,dp[i][1]表示在第i天结束后处于“可交易”状态的最大利润,dp[i][2]表示在第i天结束后处于“冷冻期”状态的最大利润。解释一下:“买入”状态表示手里有股票,“可交易”状态表示手里没有股票且不处于冷冻期,“冷冻期”表示处于冷冻期。

状态转移方程

这个题目比较复杂,需要画一个状态机图结合来看。

在这里插入图片描述

结合图可以写出方程:
dp[i][0]=max(dp[i-1][0],dp[i-1][1]-prices[i])
dp[i][1]=max(dp[i-1][2],dp[i-1][1])
dp[i][2]=dp[i-1][0]+prices[i]

初始化

根据转移方程可知必须初始化i=0的位置,dp[0][0]表示在第0天结束后处于“买入”状态,所以dp[0][0]= -prices[0]。同理dp[0][1]=0、dp[0][2]=0(第0天就冷冻期了,就说明在第一天买了又卖了,所以就是0)

填表顺序

不难发现是**从左往右,**三个表一起填

返回值

返回最后一天时,处于冷冻期和可交易状态中的较大利润,不可能最后一天处于买入状态了。

代码

class Solution {
public:
    //0:买入状态 1:可交易 2:冷冻
    int maxProfit(vector<int>& prices) {
        //创建dp表
        int len = prices.size();
        int dp[len][3];
        //初始化
        dp[0][0]=-prices[0];
        dp[0][1]=0;
        dp[0][2]=0;
        //填表
        for(int i=1;i<len;i++)
        {
            dp[i][0]=max(dp[i-1][0],dp[i-1][1]-prices[i]);
            dp[i][1]=max(dp[i-1][1],dp[i-1][2]);
            dp[i][2]=dp[i-1][0]+prices[i];
        }
        //返回
        return max(dp[len-1][1],dp[len-1][2]);
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值