代码随想录算法训练营33期第五十一天 | 309. 买卖股票的最佳时机含冷冻期\714.买卖股票的最佳时机含手续费

最新推荐文章于 2024-10-09 22:28:58 发布

Caro Rye

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量111

点赞数 1

文章标签：算法 leetcode 动态规划

本文链接：https://blog.csdn.net/weixin_43936887/article/details/138213250

版权

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

// dp当前口袋里的钱
// dp[i][0]持有股票
// dp[i][1]卖出后不持有(昨天或更早卖出股票)
// dp[i][2]今天卖出股票
// dp[i][3]冷冻期

// dp[i][0]=max(dp[i-1][0], max(dp[i-1][1]-prices[i], dp[i-1][3]-prices[i])) 保持持有或者今天购买
// dp[i][1]=max(dp[i-1][1], dp[i-1][3]-prices[i]) 上次卖后的钱和冷冻期之后的最大值
// dp[i][2]=dp[i-1][0]+prices[i];
// dp[i][3]=dp[i-1][2];

// dp[0][0]=-prices[0];
// dp[0][1]=0;
// dp[0][2]=0;
// dp[0][3]=0;

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(prices.size(), vector<int>(4, 0));

        int n = prices.size();
        if (n == 0) return 0;
        dp[0][0]=-prices[0];
        dp[0][1]=0;
        dp[0][2]=0;
        dp[0][3]=0;

        for (int i=1; i<dp.size(); i++){
            dp[i][0]=max(dp[i-1][0], max(dp[i-1][1]-prices[i], dp[i-1][3]-prices[i]));
            dp[i][1]=max(dp[i-1][1], dp[i-1][3]);
            dp[i][2]=dp[i-1][0]+prices[i];
            dp[i][3]=dp[i-1][2];
        }

        return max(dp[n - 1][3], max(dp[n - 1][1], dp[n - 1][2]));
    }
};```

714.买卖股票的最佳时机含手续费

// dp当前口袋里的钱
// dp[i][0]买入
// dp[i][1]卖出，卖出的时候扣手续费

// 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][0]+prices[i]-fee)

// dp[0][0]=-prices[i]
// dp[0][1]=0

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        vector<vector<int>> dp(prices.size(), vector<int>(2, 0));

        dp[0][0]=-prices[0];
        dp[0][1]=0;

        for (int i=1; i<dp.size(); 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][0]+prices[i]-fee);
        }

        return dp[dp.size()-1][1];
    }
};