代码随想录第九章动态规划part09 188.买卖股票的最佳时机IV 309.最佳买卖股票时机含冷冻期 714.买卖股票的最佳时机含手续费

最新推荐文章于 2024-11-01 17:00:11 发布

孩子不懂事乱写的

最新推荐文章于 2024-11-01 17:00:11 发布

阅读量163

点赞数 1

文章标签：动态规划算法

本文链接：https://blog.csdn.net/qq_53524614/article/details/142361161

版权

188.买卖股票的最佳时机IV

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        vector<int> dp(2 * k + 1, 0);
        for (int i = 0; i < k; i++){
            dp[2 * i + 1] -= prices[0];
        }
        for (int  i = 1; i < prices.size(); i++){
            for (int j = 1; j <= k; j++){
                dp[2 * j - 1] = max(dp[2 * j - 2] - prices[i], dp[2 * j - 1]);
                dp[2 * j] = max(dp[2 * j - 1] + prices[i], dp[2 * j]);
            }
        }
        return dp[2 * k];
    }
};

就是之前两次交易的扩展，吧dp数组扩大，以同样的迭代方式计算即可。

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

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.size() == 1) return 0;
        vector<vector<int>> dp(prices.size(), vector<int>(2, 0));
        dp[0][0] -= prices[0];
        dp[1][0] = max(-prices[0], -prices[1]);
        dp[1][1] = max(dp[0][1], dp[0][0] + prices[1]);
        for (int i = 2; i<prices.size(); i++){
            dp[i][0] = max(dp[i - 1][0], dp[i - 2][1] - prices[i]);
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
        }
        return dp[prices.size() - 1][1];
    }
};

冷冻期在卖出后对下次交易时间有要求，实际上只需要将迭代中中纳入计算的数据部分前推即可。

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

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        vector<int> dp(2, 0);
        dp[0] -= prices[0];
        for (int i = 1; i < prices.size(); i++){
            dp[0] = max(dp[0], dp[1] - prices[i]);
            dp[1] = max(dp[1], dp[0] + prices[i] - fee);
        }
        return dp[1];
    }
};

只要在买入的计算中把手续费算进去就行。

代码随想录第九章动态规划part09