代码随想录刷题第51天

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

太阳照常升起366

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

阅读量656

点赞数 23

文章标签：算法职场和发展 leetcode c++

本文链接：https://blog.csdn.net/weixin_74090791/article/details/136510170

版权

本文介绍了两道LeetCode题目，涉及股票买卖的最佳时机，一题包含冷冻期限制，另一题考虑了手续费。通过动态规划方法求解，分别计算了不持有股票和持有股票时的最大利润。

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

刷题即将进入尾声，一路走来颇为感慨。今天还是聚焦股市。第一题是含冷冻期的最佳买卖股票时机https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown/，本题中含有冷冻期，即当天卖出后的第2天才能重新买入，那么今天持有股票的状态就与2天前的状态有关。dp[i][0]表示第i天不持有股票对应的利润最大值，dp[i][1]表示持有股票对应的利润最大值。不持有时，可能前一天也不持有，即dp[i][0]=dp[i - 1][0]；或者前一天持有，今天卖出，也是合法的操作，dp[i][0] = dp[i-1][1] + prices[i]。若第i天持有股票，可能前一天也持有股票，dp[i][1] = dp[i-1][1]；也可能是第i-2天卖出股票，即第i-2天时不持有的状态，经历了一天的冷冻期，第i天重新买入股票，dp[i][1] = dp[i-2][0] - prices[i]。对dp数组进行初始化，dp[0][0]对应着第0天不持有股票，dp[0][0] = 0；dp[0][1]对应第0天持有股票，即在第0天买入股票，dp[0][1] = -prices[0]。dp[1][1]表示第一天持有股票，则可能在第0天或第1天购入，dp[1][1] = max(dp[0][1], -prices[1])。从前向后遍历dp数组。

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

    }
};

卡哥是划分出4种状态：持有股票、保持卖出股票的状态、今天卖出股票、冷冻期，更加细致，但是不好理解。

第二题是含手续费的买卖股票https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-transaction-fee/，就是在买卖操作上追加手续费用，秒了。

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

    }
};

股票系列结束😄