力扣309. 最佳买卖股票时机含冷冻期

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

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

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

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

示例 1:

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

示例 2:

输入: prices = [1]
输出: 0

提示:

  • 1 <= prices.length <= 5000
  • 0 <= prices[i] <= 1000

解析:

这道题加入冷冻期后,可以分成以下几种情况:

持有股票,不持有股票,以及冷冻期。其中不持有股票分成:(很久之前已经过了冷冻期,继续不持有,还有一种是刚刚过冷冻期),以及刚刚卖出股票。

其中 持有股票:首先是之前已经持有并继续持有,还有一种是之前是冷冻期,最后是之前已经卖出很久了,这三种情况取最大值。

不持有股票:很久之前已经过了冷冻期,继续不持有,以及刚刚过冷冻期取最大值。

刚刚卖出股票:之前是持有状态。

今天是冷冻期:之前是卖出状态。

源码:

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(4, vector<int>(prices.size(), 0));
        dp[3][0] = -prices[0];
        for (int i = 1; i < prices.size(); i++) {
            dp[0][i] = max(dp[0][i - 1], dp[2][i - 1]); //保持不持有的状态
            dp[1][i] = dp[3][i - 1] + prices[i]; //刚刚卖出的状态
            dp[2][i] = dp[1][i - 1];//今天是冷冻期
            dp[3][i] = max(dp[3][i - 1], max(dp[2][i - 1] - prices[i], dp[0][i - 1] - prices[i]));//持有的状态
        }
        for (int j = 0; j < 4; j++) {
            for (int i = 0; i < prices.size(); i++) {
                cout << dp[j][i] << " ";
            }
            cout << endl;
        }
        return max(dp[1][prices.size() - 1], max(dp[2][prices.size() - 1], dp[0][prices.size() - 1]));
    }
};

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值