力扣309. 最佳买卖股票时机含冷冻期(动态规划)暂时不是很懂,之后再看
给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
示例:
输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
每一天的状态有四种:
-
持有
-
抛售
-
冷冻
-
可买
假设以变量hold, sale, freeze, aval
来表示这四种状态下的最大收益,则以持有状态的收益为例:
今天持有,则昨天一定是是 持有、冷冻、可买
三种状态之一,且如果昨天是持有,我的收益要加上今天的股票涨幅,否则收益不变,即hold=max(hold+p-yesterday,freeze,aval)
,p为今天股价,yesterday为昨天股价。
显然,其他三种状态的对应关系为:
今日抛售=昨日持有+涨幅
今日冷冻=昨日抛售
今日可买=max(昨日冷冻,昨日可买)
//
// main.cpp
// 309maxProfit
//
// Created by MXQ on 2020/10/22.
//
#include <iostream>
#include<vector>
using namespace::std;
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n=prices.size();
if(n==0 || n==1)return 0;
vector<int>in(n);
vector<int>out(n);
vector<int>fraze(n);
out[0]=0;
in[0]=-prices[0];
fraze[0]=0;
for (int i=1; i<n; i++) {
//转移方程
in[i]=max(fraze[i-1]-prices[i],in[i-1]);//买入
out[i]=max(in[i-1]+prices[i],out[i-1]);//卖出
fraze[i]=out[i-1];
}
return max(in[n-1],max(out[n-1],fraze[n-1]));
}
};
int main(int argc, const char * argv[]) {
// insert code here...
vector<int>prices={1,2,3,0,2};
Solution s;
auto result=s.maxProfit(prices);
std::cout << result<<endl;
std::cout << "Hello, World!\n";
return 0;
}