「动态规划」买卖股票的最佳时机含手续费

力扣原题链接,点击跳转。

给定一个整数数组prices,其中pricesi表示第i天的股票价格,整数fee代表了交易股票的手续费用。你可以无限次地完成交易,但是你每笔交易都需要付手续费。这里的一笔交易指买入持有到卖出的整个过程,每笔交易你只需要支付一次手续费。如果你已经购买了一个股票,在卖出它之前你就不能再购买股票了。请计算你能获得的最大利润。

要分清楚状态和操作,我会用「」把状态和操作都标记出来。注意到,第i天结束后,手里有可能有股票,也就是处于「买入」状态;手里也有可能没有股票,也就是处于「卖出」状态。对于前者,接下来可以执行的操作有「持有股票」和「卖出股票」;对于后者,接下来可以执行的操作有「观望」和「买入股票」。在进行「买入股票」和「卖出股票」时会引起利润变化,前者减少利润,后者增加利润。「卖出股票」时,进行了一笔完整的交易,要缴纳手续费。

我们用动态规划的思想来解决这个问题。首先确定状态表示。我们用f表示处于「买入」状态,g表示处于「卖出」状态,i表示时间处于第i天结束时,表中的值表示最大利润。也就是说:

  • fi表示在第i天结束时,处于「买入」状态,此时能获得的最大利润。
  • gi表示在第i天结束时,处于「卖出」状态,此时能获得的最大利润。

接下来推导状态转移方程:

考虑fi。在第i天结束时处于「买入」状态,分类讨论:

  • 如果第i-1天结束时处于「买入」状态,并且在第i天「持有股票」,有fi=fi-1。
  • 如果第i-1天结束时处于「卖出」状态,并且在第i天「买入股票」,有fi=gi-1-pricesi。

考虑gi。在第i天结束时处于「卖出」状态,分类讨论:

  • 如果第i-1天结束时处于「卖出」状态,并且在第i天「观望」,有gi=gi-1。
  • 如果第i-1天结束时处于「买入」状态,并且在第i天「卖出股票」,有gi=fi-1+pricesi-fee。

综上,有:fi=max(fi-1,gi-1-pricesi),gi=max(gi-1,fi-1+pricesi-fee)。

接着考虑初始化的问题。

  • 对于f0,要在第0天结束时处于「买入」状态,就要在第0天「买入股票」,所以f0=-prices0。
  • 对于g0,要在第0天结束时处于「卖出」状态,就要在第0天「观望」,所以g0=0。

填表时应按照从左往右的顺序,同时填f表和g表。最后应返回gn-1,因为只有最后一天结束时处于「卖出」状态,才有可能获得最大利润。

cpp 复制代码
class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int n = prices.size();
        // 创建dp表
        vector<int> f(n);
        auto g = f;
        // 初始化g[0]=0
        f[0] = -prices[0];
        // 填表
        for (int i = 1; i < n; i++)
        {
            f[i] = max(f[i-1], g[i-1] - prices[i]);
            g[i] = max(g[i-1], f[i-1] + prices[i] - fee);
        }
        return g[n-1];
    }
};
相关推荐
Tairitsu_H27 分钟前
[动态规划#1] 子数组 | 最大子数组和 | 环形子数组的最大和 | 乘积最大子数组 | 乘积为正数的最长子数组
算法·动态规划
凯瑟琳.奥古斯特1 小时前
动态调优IoU提升OCR文本合并精度
开发语言·python·算法
布莱克6051 小时前
A* 算法详解:从原理到实现
开发语言·c++·算法
段一凡-华北理工大学1 小时前
高炉炼铁机器视觉与智能识别十八讲~系列文章15:可靠性篇:防误报、容错与识别系统的鲁棒性设计
人工智能·算法·机器学习·高炉大模型·高炉机器视觉·高炉炼铁模型开发
某林2122 小时前
局部路径规划器 DWB:速度采样与评分权重的取值分析
人工智能·算法·架构·机器人·技术复盘
AI职业加油站2 小时前
模数共振行动解读|大模型应用工程师证书,匹配行业 RAG/Agent 落地人才需求
人工智能·学习·算法·职场和发展·职场发展
lengxuenong3 小时前
树状数组 · 信息学学习手册2-离散化与逆序对
c++·学习·算法
琳达爱分享3 小时前
五个线程同时去换 token,四个被踢下线:刷新接口其实不是幂等的
后端·算法·架构
十二3 小时前
【洛谷题解】P1115 最大子段和(前缀和 + 维护最小前缀)
数据结构·c++·算法·蓝桥杯
renhongxia14 小时前
RAG检索增强生成:打通大模型与外部知识的通道
大数据·人工智能·算法