多状态Dp问题——买卖股票的最佳时机含冷冻期

目录

一,题目

二,题目接口

三,解题思路及其代码


一,题目

给定一个整数数组prices,其中第 prices[i] 表示第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):

  • 卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。

**注意:**你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

示例 1:

复制代码
输入: prices = [1,2,3,0,2]
输出: 3 
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

示例 2:

复制代码
输入: prices = [1]
输出: 0

二,题目接口

cpp 复制代码
class Solution {
public:
    int maxProfit(vector<int>& prices) {

    }
};

三,解题思路及其代码

首先,要明确的一点便是这道题还是一个多状态的dp问题。在这样一道题里面,在每一天都会有三种状态:**1.**今天处于卖出状态。

**2.**今天处于买入状态。

**3.**今天处于冷冻期。

在明确好这些状态以后,便可以开始列举这几种状态间的转换关系了。

转换到卖出状态的情况:1.前一天处于买入状态,今天卖出股票。3.前一天处于卖出状态,这一天什么也不做。

转换到买入状态:1.前一天处于冷冻期状态,今天买入。2.前一天处于买入状态,今天啥也不做。

转换到冷冻期:1.前一天处于卖出状态。

画出状态转移图如下:

在推理完这些状态转移关系以后便可以推导出要求最大值的情况下的状态转移方程,设:f(i),g(i),x(i)分别是卖出状态,买入状态,冷冻期的最大利润。那便可以推导出如下的状态转移方程:

cpp 复制代码
 f[i] = max(x[i-1]-prices[i],f[i-1]);
 g[i] = max(f[i-1]+prices[i],g[i-1]);
 x[i] = g[i-1];

然后在完成初始化以后便可以写出这道题的动态规划解法了,代码如下:

cpp 复制代码
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        vector<int>f(n);
        auto g = f;
        auto x = f;

        f[0]= -prices[0];//初始化

        for(int i = 1;i<n;i++)
        {
            f[i] = max(x[i-1]-prices[i],f[i-1]);
            g[i] = max(f[i-1]+prices[i],g[i-1]);
            x[i] = g[i-1];
        }

        return max(x[n-1],g[n-1]);

    }
};

过啦!!!

相关推荐
袁小皮皮不皮5 分钟前
1.HCIP BFD 学习笔记(优化版)
服务器·网络·笔记·网络协议·学习·智能路由器·ip
坚果派·白晓明13 分钟前
【鸿蒙PC】SDL3 移植:AtomCode Skills 4 步速通多媒体库适配
c++·华为·ai编程·harmonyos·atomcode·c/c++三方库
手写码匠24 分钟前
手写 GraphRAG:从零实现图增强检索增强生成系统
人工智能·深度学习·算法·aigc
装不满的克莱因瓶29 分钟前
【自动驾驶领域】学习 Cityscapes 数据集——城市街景语义理解的标准基准
人工智能·pytorch·python·深度学习·学习·机器学习·自动驾驶
BomanGe130 分钟前
NSK重载高刚性滚珠丝杠技术详解
经验分享·算法·规格说明书
赴生-1 小时前
C++进阶 C++11(下)
开发语言·c++
有点。1 小时前
C++(贪心算法一)
c++·贪心算法
Matrix_111 小时前
手机里的计算摄影:广角形变校正算法
人工智能·算法·智能手机·计算摄影
清辞8531 小时前
产品经理需求推进流程
大数据·深度学习·学习·产品经理
WBluuue1 小时前
数据结构与算法:有序表(二):跳表
数据结构·c++·算法·skiplist