C++ day51 买卖股票最佳时期
题目1:309 买卖股票的最佳时机含冷冻期
题目链接:买卖股票的最佳时机含冷冻期
对题目的理解
prices[i]表示第i天股票的价格,尽可能多地完成更多的交易,不能同时进行多笔交易,卖出股票后,第二天无法买入股票(冷冻期是1天),计算最大利润
动态规划
动规五部曲
1)dp数组及下标i的含义
状态1:dp[i][0] 第i天持有股票的状态最大现金
状态2:dp[i][1] 第i天保持股票卖出的状态,冷冻期之后最大现金
状态3:dp[i][2] 第i天具体卖出股票的状态,冷冻期的前一天最大现金
状态4:dp[i][3] 第i天冷冻期的状态最大现金
最终求解:dp[prices.size()-1][1] dp[prices.size()-1][2] dp[prices.size()-1][3]的
2)递推公式
dp[i][0] = dp[i-1][0] 前一天持有股票
买入股票:前一天是冷冻期;前一天保持卖出股票状态
dp[i][0] = dp[i-1][3]-prices[i]
dp[i][0] = dp[i-1][1]-prices[i]
dp[i][0] = max(dp[i-1][0],dp[i-1][3]-prices[i],dp[i-1][1]-prices[i])
dp[i][1] = dp[i-1][1] 冷冻期之后一直保持股票卖出的状态
dp[i][1] = dp[i-1][3] 冷冻期的状态的下一天
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] 前一天是具体卖出股票的状态
3)dp数组初始化
dp[0][0]=-prices[0]
dp[0][1]的状态是一个非法状态,看递推公式中需要初始化为多少,dp[1][0]=dp[0][1]-prices[1],所以将dp[0][1]初始化成0(当天买当天卖)
dp[1][0] = dp[0][3]-prices[1] 所以将dp[0][3]初始化为0
同理将dp[0][2]初始化为0
4)遍历顺序
根据递推公式,dp[i] 依赖于 dp[i-1],从小到大遍历
5)打印dp数组
代码
class Solution {
public:
int maxProfit(vector<int>& prices) {
//定义dp数组
vector<vector<int>> dp(prices.size(),vector<int>(4,0));
//初始化dp数组
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][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]));
}
};
- 时间复杂度:O(n)
- 空间复杂度:O(n)
题目2:714 买卖股票的最佳时机含手续费
题目链接:买卖股票的最佳时机含手续费·
对题目的理解
prices[i]表示第i天的股票价格 fee代表交易股票的手续费,每笔交易都需要附手续费,可以无限次的进行交易,不能同时进行多次交易,返回利润的最大值。
动态规划
动规五部曲
1)dp数组及下标i的含义
dp[i][0]:第i天持有股票的最大现金数
dp[i][1]:第i天不持有股票的最大现金数
最后求解:dp[prices.size()-1][1]
2)递推公式
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),不持有股票可以分为两种情况,一种是一直不持有,还有一种是前一天持有股票,那么,今天卖出,就达到了不持有的情况,同时需要减去手续费
3)dp数组初始化
根据递推公式,初始化dp[0][0]和dp[0][1],
dp[0][0]代表第i天持有股票的最大现金数,dp[1][0]=dp[0][0],那么为-prices[0]
dp[1][0]代表第i天不持有股票的最大现金数 dp[1][1]=dp[0][1] 那么第一天一定是买或者不买股票,所以不持有股票的最大现金数是0
4)遍历顺序
根据递推公式,从小到大遍历 for(i=1;i<prices.size();i++)注意i从1开始遍历
5)打印dp数组
class Solution {
public:
int maxProfit(vector<int>& prices, int fee) {
//定义dp数组
vector<vector<int>> dp(prices.size(),vector<int>(2,0));
//初始化dp数组
dp[0][0] = -prices[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];
}
};
- 时间复杂度:O(n)
- 空间复杂度:O(n)