因为这里有一个冷冻期,这就十分不好处理。
所以我们用 f i , 0 f_{i,0} fi,0 表示当前买来了股票, f i , 1 f_{i,1} fi,1 表示当前没有股票,处于冷冻期, f i , 2 f_{i,2} fi,2表示当前没有骨片也不处于冷冻期。(这个不一定是要在 i i i 这个点满足条件,也可以在 1 → i 1\to i 1→i 之间某个值满足这个东西)
转移方程显然是
f[i][0]=max(f[i-1][2]-a[i],f[i-1][0]);
f[i][1]=max(f[i-1][0]+a[i],f[i-1][1]);
f[i][2]=max(f[i-1][1],f[i-1][2]);
所以代码就是
class Solution {
public:
int n,f[100001][3];
int maxProfit(vector<int>& a) {
if(a.empty()){
return 0;
}
n=a.size();
f[0][0]=-a[0];
for(int i=1;i<n;i++){
f[i][0]=max(f[i-1][2]-a[i],f[i-1][0]);
f[i][1]=max(f[i-1][0]+a[i],f[i-1][1]);
f[i][2]=max(f[i-1][1],f[i-1][2]);
}
return max(f[n-1][1],f[n-1][2]);
}
};