今天下午继续补动态规划的题目,股票问题马上就要搞完了,开冲!!!
今日任务:
- 309.最佳买卖股票时机含冷冻期
- 714.买卖股票的最佳时机含手续费
题目一:309.最佳买卖股票时机含冷冻期
Leetcode题目:【309.最佳买卖股票时机含冷冻期】
这道题目相当于对买卖股票II多了一个冷冻期。卖出股票之后需要一天冷冻期,才能买入股票。按照动规五部曲开始:
1、确定dp数组的含义(只需要4个状态就可以把时间描述清楚了):
dp[i][0]:持股;
dp[i][1]:保持卖出股票的状态;
dp[i][2]:具体卖出股票的状态(具体在那一天中卖出股票,因为有冷冻期的存在,所以要将不持股的状态拆分开);
dp[i][3]:冷冻期的状态。
2、确定递推公式:
(1)dp[i][0] 包括三种:之前买入的:保持持股的状态,就是这次买入的,包括冷冻期之后买入和保持卖出状态后的买入。
所以dp[i][0] = max(dp[i-1][0], dp[i-1][3] -prices[i], dp[i-1][1] -prices[i]);
(2)dp[i][1] :包括一直保持卖出股票的状态和冷冻区:dp[i-1][1] , dp[i-1][3]
(3)dp[i][2] = dp[i-1][0] + price[i];
(4)dp[i][3] = dp[i-1][2];
3、初始化:
dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[0][2] = 0;// 理解为当天买,当天卖
dp[0][3] = 0;
注意:如何使用max进行三个数的比较
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(), vector<int>(4));
dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[0][2] = 0;
dp[0][3] = 0;
for(int i = 1; i<prices.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[prices.size()-1][1], max(dp[prices.size()-1][2], dp[prices.size()-1][3]));
}
};
题目二:714.买卖股票的最佳时机含手续费
Leetcode题目:【714.买卖股票的最佳时机含手续费】
这个就是股票II然后多加了一个手续费,其实就是在买入股票的时候加一个手续费即可,按照动规五部曲:
(1)确定dp数组的含义:
dp[i][0]:表示持股的状态,所获得金钱数;
dp[i][1]:表示非持股状态,所获得金钱数;
(2)确定递推公式:
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]);
(3)初始化:
dp[0][0] = prices[0];
dp[0][1] = 0;
(4)遍历顺序:从左到右遍历即可;
(5)打印dp数组
class Solution {
public:
int maxProfit(vector<int>& prices, int fee) {
vector<vector<int>> dp(prices.size(), vector<int>(2));
dp[0][0] = -prices[0];
dp[0][1] = 0;
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][1];
}
};