题目链接:leetcode309
思路:因为本题中规定买入,卖出和冷冻操作,规定上一次卖出和下一次买入之间需要至少一天的冷冻期,因此我们定义三种状态sell、buy和cooldown,分别对应了到第i天为止最后一个操作是买入、卖出和冷冻所对应的最大利润,则状态转移方程如下:
- sell[i]=max(sell[i-1],buy[i-1]+prices[i]);
- buy[i]=max(buy[i-1],cooldown[i-1]-prices[i]);
- cooldown[i]=max(cooldown[i-1],max(sell[i-1],buy[i-1])).
Code:
class Solution {
public:
int maxProfit(vector<int>& prices) {
if(prices.empty()) return 0;
int n=prices.size(); //数量
vector<int> sell(n,0); //手中不持有股票
vector<int> buy(n,0); //手中持有股票
vector<int> cooldown(n,0); //此时为冷冻期(可能持有也可能不持有)的最大利润
buy[0]=-prices[0]; //根据定义第一天持有股票就一定要购买
for(int i=1;i<n;i++){
sell[i]=max(sell[i-1],buy[i-1]+prices[i]); //不持有股票要
buy[i]=max(buy[i-1],cooldown[i-1]-prices[i]); //持有股票由于有冷冻期的原因要经过一个冷冻期才能够再次购买
cooldown[i]=max(cooldown[i-1],max(sell[i-1],buy[i-1])); //更新冷冻期的最大利润
}
return sell.back();
}
};