309. 买卖股票的最佳时机含冷冻期(题目链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台)
思路:增加了一个冷冻期,原本的二维数组dp的第二维的长度就要从2变为4,依次代表:0持有股票的状态、1从很早开始就不持有股票的状态、2今天刚把股票卖掉、3昨天刚把股票卖掉,今天为冷冻期。其实也可以将1状态和2状态合为一个状态,那么碰到今天本不持有股票状态(情况2带来的结果),想买入一只股票时,并不违法。(好像按照规定卖完一只股票后不能立马又买入一支股票,但是看运算结果我们就会发现对应+prices[i]-prices[i],说明通过这样来持有的股票,dp[i][0]=dp[i-1][0])。不过分成四种情况来处理更容易理解。
int maxProfit(vector<int>& prices) {
int size = prices.size();
if(size==1) return 0;
vector<vector<int>> dp(size, vector<int>(4, 0));
dp[0][0] = -prices[0];
for(int i=1; i<size; i++){
dp[i][0] = max(dp[i-1][0], max(dp[i-1][1]-prices[i], dp[i-1][3]-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[size-1][1], max(dp[size-1][2], dp[size-1][3]));
}
714. 买卖股票的最佳时机含手续费(题目链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台)
思路:和前几天做的普通的无限次买卖股票基本无异,只需要在每次卖出股票时减去fee。
int maxProfit(vector<int>& prices, int fee) {
int size = prices.size();
if(size==1) return 0;
vector<vector<int>> dp(size, vector<int>(2, 0));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for(int i=1; i<size; i++){
dp[i][0] = max(dp[i-1][0], dp[i-1][1]-prices[i]);
dp[i][1] = max(dp[i-1][0]+prices[i]-fee, dp[i-1][1]);
}
return dp[size-1][1];
}