leetcode 714 买卖股票的最佳时机含手续费(dp)

题目描述:

  给定一个整数数组 prices,其中第 i 个元素代表了第 i 天的股票价格 ;非负整数 fee 代表了交易股票的手续费用。你可以无限次地完成交易,但是你每笔交易都需要付手续费。如果你已经购买了一个股票,在卖出它之前你就不能再继续购买股票了。

返回获得利润的最大值。注意:这里的一笔交易指买入持有并卖出股票的整个过程,每笔交易你只需要为支付一次手续费。

题解:

  类似的股票问题都可以通过定义一个$dp[i][j][k]$来描述,其中$i$为第$i$天、$j$为允许购买的次数、$k$为状态。状态的定义一般为是否持有股票,这道题目没有购买次数的限制,所以一般的二维dp就够用了,定义$k == 0$的时候不持有股票、$k == 1$的时候持有股票,状态转移如下:

  $dp[i][0] = max(dp[i-1][0],dp[i-1][1]-prices[i]) $ // 这里$prices[i]$为第$i$天股票的一个交易额,对于第$i$天不持有股票的情况,从第$i-1$天不持有股票、第$i-1$天持有股票转移过来就可以了

  $dp[i][1] = max(dp[i-1][1],max(dp[i-1][0] + prices[i] - fee,dp[i][0] + prices[i] - fee))$ //,对于第$i$天持有股票的情况,从第$i-1$天不持有股票、第$i-1$天持有股票以及第$i$天不持有股票转移过去就可以了

  二维的$dp$数组还是太占用内存,可以简化一下。由于状态的转移只在相邻的天数进行,实际上我们用两个变量就可以表示所有的状态了,具体见代码~

AC代码:

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int cash = 0;
        int hold = -prices[0]; // 初始化
        // cash 表示不持有  hold表示持有
        int len = prices.size();
        if(len == 0) return 0;
        for(int i = 0;i<len;i++)
        {
            cash = max(cash,hold + prices[i] - fee);
            hold = max(hold,cash - prices[i]); // 这里的cash 等效于 max(dp[i-1][0] + prices[i] - fee,dp[i][0] + prices[i] - fee)
        }
        return cash;
    }
};

推荐一下这个股票问题的一个小结(https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-transaction-fee/solution/yi-ge-fang-fa-tuan-mie-6-dao-gu-piao-wen-ti-by-l-2/

猜你喜欢

转载自www.cnblogs.com/z1141000271/p/12904067.html
今日推荐