309.最佳买卖股票时机含冷冻期
这道题主要就是搞懂dp数组含义以及状态之间的转换,没看答案能自己做出来
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(),vector<int>(5,0));
//前两天之前卖出股票,不持有股票的最大现金
dp[0][0] = 0;
//持有股票的最大现金
dp[0][1] = -prices[0];
//冷冻期
dp[0][3] = 0;
//今天卖出股票
dp[0][4] = 0;
for(int i = 1;i<prices.size();i++){
dp[i][0] = max(dp[i-1][0],dp[i-1][3]);
dp[i][1] = max(dp[i-1][1],max(dp[i-1][3]-prices[i],dp[i-1][0]-prices[i]));
dp[i][3] = dp[i-1][4];
dp[i][4] = dp[i-1][1] + prices[i];
}
return max(dp[prices.size()-1][0],max(dp[prices.size()-1][3],dp[prices.size()-1][4]));
}
};
714.买卖股票的最佳时机含手续费
这道题和买卖股票二一样,无非就是再递推公式中减去fee
class Solution {
public:
int maxProfit(vector<int>& prices, int fee) {
vector<vector<int>> dp(prices.size(),vector<int>(2,0));
dp[0][0] = 0;//不持有
dp[0][1] = -prices[0];//持有
for(int i = 1;i<prices.size();i++){
dp[i][0] = max(dp[i-1][0],dp[i-1][1]+prices[i]-fee);
dp[i][1] = max(dp[i-1][1],dp[i-1][0]-prices[i]);
}
return dp[prices.size()-1][0];
}
};
总结
股票问题 最难的就是搞懂dp数组含义和递推公式