代码随想录算法训练营day51

最新推荐文章于 2024-07-12 18:56:03 发布

Mitakiyuki

最新推荐文章于 2024-07-12 18:56:03 发布

阅读量276

点赞数 2

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

本文链接：https://blog.csdn.net/Mitakiyuki/article/details/138196779

版权

文章介绍了使用动态规划方法解决股票买卖问题，包括两种情况：一种考虑冷冻期，dp数组有四个状态；另一种考虑手续费，dp数组有两个状态。通过前向遍历计算持有/不持有股票的最大收益。

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

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

五部曲：

dp数组下标及含义：

dp[i][0] 表示持有股票；
dp[i][1]表示保持卖出股票的状态；
dp[i][2]表示今天卖出股票；
dp[i][3]表示冷冻期状态；

dp数组初始化：

dp[i][0]=-prices[0]；
dp[i][1]=0；
dp[i][2]=0；
dp[i][3]=0；

递推公式：

dp[i][0] = max(dp[i - 1][0], dp[i - 1][3] - prices[i], dp[i - 1][1] - 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];

遍历方向：从前到后遍历

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if (n == 0)
            return 0;
        vector<vector<int>> dp(n, vector<int>(4, 0));
        dp[0][0] -= prices[0];
        for (int i = 1; i < n; i++) {
            dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3] - prices[i],dp[i - 1][1] - 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数组初始化：

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

递推公式：

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)；

遍历方向：从前到后遍历

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][0], dp[n - 1][1]);
    }
};