多状态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]);

    }
};

过啦!!!

相关推荐
zyl7653 小时前
FlashAttention 详解:从 GPU 硬件、普通 Attention 到分块计算与 Online Softmax
算法
爱读书的小胖7 小时前
LeetCode 122:买卖股票的最佳时机 II(C)解题分析
算法
结局无敌7 小时前
React Native for OpenHarmony 实战:三方库 react-native-modal-dialog 的鸿蒙化适配指南
深度学习·算法·react native·react.js·harmonyos
OPEN-F8 小时前
ROS2系列教程:上位机建图与导航(原理综合)
算法·机器人
weixin_307779138 小时前
C++代码实现MATLAB中的fitrtree函数功能
开发语言·c++·算法·matlab
xier_ran8 小时前
【infra之路】AI 编译器学习笔记(十三):AI 编译器全景——TVM、XLA、IREE 与其他
人工智能·笔记·学习
连山齐名8 小时前
ai应用开发的学习笔记
服务器·笔记·学习
AI wins8 小时前
c++新增頭文件
c++
Shadow(⊙o⊙)8 小时前
仿muduo实现高并发服务器——事件就绪管理Channel.hpp
服务器·c++
tomwang03228 小时前
路径优化建模、优化算法分类及工程化小结
算法·数学建模·分类