代码随想录第四十九天、第五十天、第五十一天|股票买卖问题
Leetcode121. 买卖股票的最佳时机
题目链接:Leetcode121. 买卖股票的最佳时机
非常简单的简单题,dp我都懒得用
class Solution {
public:
int maxProfit(vector<int>& prices) {
int low = INT_MAX;
int result = 0;
for (int i = 0; i < prices.size(); i++) {
low = min(low, prices[i]); // 取最左最小价格
result = max(result, prices[i] - low); // 直接取最大区间利润
}
return result;
}
};
Leetcode122.买卖股票的最佳时机II
题目链接:Leetcode122.买卖股票的最佳时机II
这道题之前用贪心做过,见高就卖,见低就换就行了,非常好用,再试试dp版本
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.size()==1) return 0;
int dp[prices.size()][2];
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]);
}
};
Leetcode123.买卖股票的最佳时机III
题目链接:Leetcode123.买卖股票的最佳时机III
这题确实有点难度,不过做过上一题之后思路就很一致,属于是hard题里的简单题了吧。梳理清楚情况
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.size()==1) return 0;
int dp[prices.size()][5];
dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0;
dp[0][3]=-prices[0];
dp[0][4]=0;
for(int i=1;i<prices.size();i++){
dp[i][0]=dp[i-1][0];//啥也不做
dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1]);//首次持有股票
dp[i][2]=max(dp[i-1][1]+prices[i],dp[i-1][2]);//首次卖出股票
dp[i][3]=max(dp[i-1][2]-prices[i],dp[i-1][3]);//二次持有股票
dp[i][4]=max(dp[i-1][3]+prices[i],dp[i-1][4]);//二次卖出股票
}
return dp[prices.size()-1][4];
}
};
Leetcode188.买卖股票的最佳时机IV
题目链接:Leetcode188.买卖股票的最佳时机IV
这个就是三的升级版
class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
if (prices.size() == 0||prices.size() == 1) return 0;
vector<vector<int>> dp(prices.size(), vector<int>(2 * k + 1, 0));
for (int j = 1; j < 2 * k; j += 2) {
dp[0][j] = -prices[0];
}
for (int i = 1;i < prices.size(); i++) {
for (int j = 0; j < 2 * k - 1; j += 2) {
dp[i][j + 1] = max(dp[i - 1][j + 1], dp[i - 1][j] - prices[i]);
dp[i][j + 2] = max(dp[i - 1][j + 2], dp[i - 1][j + 1] + prices[i]);
}
}
return dp[prices.size() - 1][2 * k];
}
};
Leetcode309.最佳买卖股票时机含冷冻期
题目链接:Leetcode309.最佳买卖股票时机含冷冻期
这道题已经完全理解了,纯自己做出来的。
class Solution {
public:
int maxProfit(vector<int>& prices) {
int len=prices.size();
if(len==1) return 0;
int dp[len][3];
dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0;
for(int i=1;i<len;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][2]);
dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1]);
dp[i][2]=max(dp[i-1][1]+prices[i],dp[i-1][2]);
}
return max(dp[len-1][0],max(dp[len-1][1],dp[len-1][2]));
}
};
Leetcode714.买卖股票的最佳时机含手续费
题目链接:Leetcode714.买卖股票的最佳时机含手续费
easy
class Solution {
public:
int maxProfit(vector<int>& prices, int fee) {
int len=prices.size();
if(len==1) return 0;
int dp[len][2];
dp[0][0]=0;
dp[0][1]=-prices[0]-fee;
for(int i=1;i<len;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i]);
dp[i][1]=max(dp[i-1][0]-prices[i]-fee,dp[i-1][1]);
}
return max(dp[len-1][0],dp[len-1][1]);
}
};
股票问题总结
做了这么多股票问题,其实都是逐渐复杂化的变种。
总体来说,做法都是二维动归,dp[i][n]
根据题目要求来定n有几个,大概分成当天是否持有股票,当天是第几次买卖,这几类,其他的都没啥了。
可以拿|||举个例:
dp[i][0]=dp[i-1][0];//啥也不做
dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1]);//首次持有股票
dp[i][2]=max(dp[i-1][1]+prices[i],dp[i-1][2]);//首次卖出股票
dp[i][3]=max(dp[i-1][2]-prices[i],dp[i-1][3]);//二次持有股票
dp[i][4]=max(dp[i-1][3]+prices[i],dp[i-1][4]);//二次卖出股票
梳理清楚情况就OK