代码随想录算法训练营第五十一天| ● 309.最佳买卖股票时机含冷冻期 ● 714.买卖股票的最佳时机含手续费 ●总结
最佳买卖股票时机含冷冻期
题目链接：309. 买卖股票的最佳时机含冷冻期 - 力扣（LeetCode）
思路：本题难点在于如何将冷冻期加入到状态转移方程中，不妨画个图：
按理来说，如何我们正处于买入状态，将股票卖出后，应该是冷冻状态，但是这里多加了一个今日卖出状态，就是将今日卖出和卖出状态分开看，避免两个状态（买入态和冷动态）之间相互转化，否则容易把自己绕进去了。
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(),vector<int>(4));
dp[0][0]=-prices[0];
for(int i=1;i<prices.size();i++){
/*持有态*/ dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3] - prices[i], dp[i - 1][1] - 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[prices.size()-1][3],max(dp[prices.size()-1][2],dp[prices.size()-1][1]));
}
};好像三个状态也挺好理解的
class Solution {
public:
int maxProfit(vector<int>& prices) {
if (prices.empty()) {
return 0;
}
int n = prices.size();
// f[i][0]: 手上持有股票的最大收益
// f[i][1]: 手上不持有股票，并且处于冷冻期中的累计最大收益
// f[i][2]: 手上不持有股票，并且不在冷冻期中的累计最大收益
vector<vector<int>> f(n, vector<int>(3));
f[0][0] = -prices[0];
for (int i = 1; i < n; ++i) {
f[i][0] = max(f[i - 1][0], f[i - 1][2] - prices[i]);
f[i][1] = f[i - 1][0] + prices[i];
f[i][2] = max(f[i - 1][1], f[i - 1][2]);
}
return max(f[n - 1][1], f[n - 1][2]);
}
};买卖股票的最佳时机含手续费
题目链接：714. 买卖股票的最佳时机含手续费 - 力扣（LeetCode）
思路：别忘了初始化的时候也有要收手续费
class Solution {
public:
int maxProfit(vector<int>& prices, int fee) {
vector<vector<int>> dp(prices.size(),vector<int>(2));
dp[0][1]=-(prices[0]+fee);
for(int i=1;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-1][0]-prices[i]-fee);
}
return dp[prices.size()-1][0];
}
};
