思路
状态机dp
这里的前一天均指昨天:
前一天可能出先的三种状态 卖出、冷冻期、买入
若昨天是冷冻期,
- 今天可以买入,这时候买入花费-p[i]
- 今天可继续冷冻,不花钱 0
昨天已买入
- 今天可卖出,+p[i]
- 今天可保持不卖出, 不花钱 0
昨天卖出
- 今天进入冷冻期限,不买卖 0
综上所述,一天里面的三种状态,可以由昨天的状态得到用0, 1, 2分别表示三种状态:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
vector<vector<int>>f (n , vector<int>(3, -1e8));
f[0][0] = 0, f[0][1] = -prices[0]; //初始化第一天,可以不买,也可以卖出
for (int i = 1; i < n; i++) {
f[i][0] = max(f[i - 1][0], f[i - 1][2]);
f[i][1] = max(f[i - 1][0] - prices[i], f[i - 1][1]);
f[i][2] = f[i - 1][1] + prices[i];
}
return max(f[n - 1][0], f[n - 1][2]); //最大价格时一定已经卖出
}
};