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

最新推荐文章于 2024-09-14 18:52:56 发布

Hanknet

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量46

点赞数

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

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

版权

leetCode 专栏收录该内容

50 篇文章 0 订阅

订阅专栏

文章介绍了两种使用动态规划解决买卖股票最佳时机的问题。在问题123中，状态dp包括不操作、持有和未持有的状态，通过遍历股票价格更新状态，最终得到最大利润。问题188扩展到可以进行k次交易，状态dp表示第i天不同持有状态下的最大利润，同样利用动态规划求解，时间复杂度与空间复杂度分别与问题规模和交易次数相关。

摘要由CSDN通过智能技术生成

123. 买卖股票的最佳时机 III

方法：状态dp

状态表示：0不操作1第一次持有2第一次未持有3第二次持有4第二次未持有

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

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

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

方法：状态dp

状态表示：dp[i][j]表示第i天状态为j的利润的集合属性：最大值

0 表示不操作

1 第一次买入

2 第一次卖出

3 第二次买入

4 第二次卖出

.....

总结规律得出：j为0表示不操作为奇数时表示持有，为偶数时表示未持有

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

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