提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
提示:这里可以添加本文要记录的大概内容:
今天是跟着代码随想录刷题的第46天,主要是学了买卖股票的最佳时机三个问题
提示:以下是本篇文章正文内容,下面案例可供参考
一、买卖股票的最佳时机
思路:dp[i][0]是第i天不持有股最多有多少钱,dp[i][1]是第i天持有股最多有多少钱
进行动态规划,最后看最后一天持股或者不持股哪个钱多
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(),vector<int>(2));
//dp[i][0]是第i天不持有股最多有多少钱,dp[i][1]是第i天持有股最多有多少钱
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]);//今天不持股,保持了昨天也不持或者昨天持股,今天卖了
dp[i][1]=max(dp[i-1][1],-prices[i]);//今天持股,保持的昨天持股,或者今天刚买的,因为只能买一次,所以是-prices[i]
}
return max(dp[prices.size()-1][0],dp[prices.size()-1][1]);
}
};
二、买卖股票的最佳时机2
思路:就是在今天持股最多有多少钱那里不一样,如果今天持股计算最多前需要算上,今天刚买的,但是这种情况是昨天不持股的价格减去今天购买的价格,因为可以无限次购买
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(),vector<int>(2));
//dp[i][0]是第i天不持有股最多有多少钱,dp[i][1]是第i天持有股最多有多少钱
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]);//今天不持股,保持了昨天也不持或者昨天持股,今天卖了
dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]);//今天持股,保持的昨天持股,或者今天刚买的,但是这种情况是昨天不持股的价格减去今天购买的价格
}
return max(dp[prices.size()-1][0],dp[prices.size()-1][1]);
}
};
三、买卖股票的最佳时机3
思路:dp[i][0]是第i天第一次不持有股最多有多少钱,dp[i][1]是第i天第一次持有股最多有多少钱,dp[i][2]是第i天第二次不持有股最多有多少钱,dp[i][3]是第i天第二次持有股最多有多少钱
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(),vector<int>(4));
//dp[i][0]是第i天第一次不持有股最多有多少钱,dp[i][1]是第i天第一次持有股最多有多少钱,dp[i][2]是第i天第二次不持有股最多有多少钱,dp[i][3]是第i天第二次持有股最多有多少钱
dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0;//考虑第0天,买了又卖又买又卖
dp[0][3]=-prices[0];//考虑第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],-prices[i]);//今天持股,保持的昨天持股,或者今天刚买的,注意第一次持股直接是-price[i]
dp[i][2]=max(dp[i-1][2],dp[i-1][3]+prices[i]);//今天第二次不持股,保持了昨天也是或者昨天第二次持股,今天卖了
dp[i][3]=max(dp[i-1][3],dp[i-1][0]-prices[i]);//今天第二次持股,可能保持了昨天也可能昨天第一次刚不持股,然后立马买了
}
return max(max(max(dp[prices.size()-1][0],dp[prices.size()-1][1]),dp[prices.size()-1][2]),dp[prices.size()-1][3]);
}
};