LeetCode309. 最佳买卖股票时机含冷冻期、 LeetCode714. 买卖股票的最佳时机含手续费

最新推荐文章于 2024-05-13 21:32:22 发布

Hanknet

最新推荐文章于 2024-05-13 21:32:22 发布

阅读量57

点赞数

分类专栏： algorithm 文章标签：算法动态规划 c++ leetcode Powered by 金山文档

本文链接：https://blog.csdn.net/Hanknet/article/details/129354404

版权

algorithm 专栏收录该内容

50 篇文章 0 订阅

订阅专栏

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

方法：状态dp

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

$时间复杂度O(n)，空间复杂度O(1)

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

方法：贪心

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int res = 0, n = prices.size(), minv = prices[0];
        for (int i = 1; i < n; ++i) {
            if (minv > prices[i]) minv = prices[i];
            if (prices[i] >= minv && prices[i] <= minv + fee) continue;
            if (prices[i] > minv + fee) {
                res += prices[i] - minv - fee;
                minv = prices[i] - fee;
            }
        }
        return res;
    }
};

$时间复杂度O(n)，空间复杂度O(1)

方法：状态dp

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

$时间复杂度O(n)，空间复杂度O(n)