问题拆解:
对于股票,每天只有两种状态:手中有股票,手中无股票。
对于手中有股票进行拆解:
1. 前一天手中有股票,今天什么都不做
2. 前一天手中无股票,并且前一天没有卖股票,今天买股票
对于手中无股票进行拆解:
1. 前一天手中无股票,今天什么都不做
2. 前一天手中有股票,今天卖股票
因此可以设:
dpi0表示当天手中有股票的最大获利
dpi1表示当天手中无股票的最大获利
dpi2表示当天手中无股票且前一天未卖股票的最大获利(必须要有dpi2,因为dpi1是否能买股票取决于dpi2))
12.可得状态方程如下:
dpi0 = max(dpi0, dpi2 - prices[i]); // 此处减去prices[i]表示买股票的花费,后续卖股票的获利直接加上prices[i]就行,这样就无需在卖股票的时候关心在哪里买的
dpi1 = max(dpi1, dpi0 + prices[i]); // 注意实际编码过程中dpi0被覆盖了,状态转移的顺序需要调换
dpi2 = dpi1;
最后获利的最大值为max(dpi1, dpi2);
代码块
class Solution {
public:
int maxProfit(vector<int>& prices) {
// dpi0表示第i天手中持有股票的最大获利
// dpi1表示第i天手中无股票的最大获利
// dpi2表示第i天手中无股票且不卖出股票的最大获利
int n = prices.size();
int dpi0 = INT_MIN, dpi1 = 0, dpi2 = 0;
for (int i = 0; i < n; ++i) {
int tmp = dpi0;
dpi0 = max(dpi0, dpi2 - prices[i]);
dpi2 = dpi1;
dpi1 = max(dpi1, tmp + prices[i]);
}
return max(dpi1, dpi2);
}
};
class Solution { public: int maxProfit(vector<int>& prices) { int n = prices.size(); if (!n)return 0; vector<int>buy(n), sell(n); buy[0] = -prices[0]; sell[0] = 0; for (int i = 1; i < n; ++i) { sell[i] = max(sell[i - 1], buy[i - 1] + prices[i]); if (i < 2) buy[i] = max(buy[i - 1], -prices[i]); else buy[i] = max(buy[i - 1], sell[i - 2] - prices[i]); } return sell[n - 1]; } };