代码随想录训练营第50天|309.最佳买卖股票时机含冷冻期、714. 买卖股票的最佳时机含手续费

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

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

对于买卖股票最佳时机含冷冻期，我们可以分析一下，一天中最多有多少个状态。
首先，在一天的状态，①可以购买股票，②可以卖出股票，③可能处于冷冻期，④也可能没有持有股票同时也没有购买股票也不是冷冻期（记为持续卖出股票）
因此我们用dp数组记录这四种状态。
dp[ii][0]表示购买了股票
dp[ii][1]表示卖出股票
dp[ii][2]表示冷冻期
dp[ii][3]表示持续卖出股票
他们之间的推导关系为

①如果某天处于购买股票的状态，则他可能继承了前一天购买股票的状态，也可能是由冷冻期状态或者持续卖出股票状态买入的股票。

dp[ii][0] = max(dp[ii-1][0],max(dp[ii-1][2],dp[ii-1][3])-prices[ii]);

②卖出股票状态，则是由买入股票状态推导而来，即买入股票状态按照股票价格进行卖出。

dp[ii][1] = dp[ii-1][0]+prices[ii];

③冷冻期是继承了卖出股票状态，是一个中间量，表示冷冻。

dp[ii][2] = dp[ii-1][1];

④持续卖出股票状态，只能是从冷冻期获得或者继承前一天的持续卖出股票状态。

dp[ii][3] = max(dp[ii-1][2],dp[ii-1][3]);

初始化，第一天可以买入股票，但是其他状态在含义上都是非法状态，但是根据递推公式，我们需要对这些非法状态进行赋值0。而买入股票为第一天股票的价格的相反数。

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        vector<vector<int>>dp(len,vector<int>(4,0));
        //0持有股票1保持卖出股票2卖出股票3冷冻期
        dp[0][0] = -prices[0];
        for(int ii =1;ii<len;ii++)
        {
            dp[ii][0] = max(dp[ii-1][0],max(dp[ii-1][1],dp[ii-1][3])-prices[ii]);
            dp[ii][1] = max(dp[ii-1][1],dp[ii-1][3]);
            dp[ii][2] = dp[ii-1][0]+prices[ii];
            dp[ii][3] = dp[ii-1][2];
        }
        return max(dp[len-1][1],max(dp[len-1][2],dp[len-1][3]));
    }
};

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

对于买卖股票含手续费，和前面的买卖股票Ⅱ类似，我们只需要在卖出股票的时候，减去手续费即可，无需多余的操作。
即更新数组时，只需要

//0持有股票，1不持有股票
			dp[ii][0] = max(dp[ii-1][0],dp[ii-1][1]-prices[ii]);
            dp[ii][1] = max(dp[ii-1][1],dp[ii-1][0]+prices[ii]-fee);

对此，我们可以依据之前的思路写出代码

代码

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        //0 持有股票1.不持有股票
        int len = prices.size();
        if(len ==1 ) return 0;
        vector<vector<int>>dp(len,vector<int>(2,0));
        dp[0][0] = -prices[0];
        for(int ii =1;ii<len;ii++)
        {
            dp[ii][0] = max(dp[ii-1][0],dp[ii-1][1]-prices[ii]);
            dp[ii][1] = max(dp[ii-1][1],dp[ii-1][0]+prices[ii]-fee);
        }
        return dp[len-1][1];
    }
};