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];
}
};