算法训练营day51_动态规划(3.23)
309.最佳买卖股票时机含冷冻期
相对于上面的,多了个冷冻期,加进去这个状态;
冷冻期本质上来说就是从不持有里分出来的,为不持有的第1天,而不持有相当于不持有的>=2天;
原先的卖出是持有->不持有,现在变为持有->不持有第一天->不持有第二天;
买入还是不持有->持有,因为这里不持有是不持有的>=2天,可以买入;
class Solution {
public:
int f[5010][3];
int maxProfit(vector<int>& prices) {
memset(f,-0x3f,sizeof f);
f[0][0]=0;
int n=prices.size();
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][1],f[i-1][0]-prices[i-1]);
f[i][2]=f[i-1][1]+prices[i-1];
}
return max(f[n][0],f[n][2]);
}
};
714.买入股票的最佳时机含手续费
股票2的代码,卖出的时候减去个fee即可A;
class Solution {
public:
int maxProfit(vector<int>& prices, int fee) {
int n=prices.size();
int ans=0;
vector<vector<int>> f(n+1,vector<int>(2,0));
f[0][0]=-1e5;
for(int i=1;i<=prices.size();i++){
f[i][0]=max(f[i-1][0],f[i-1][1]-prices[i-1]);
f[i][1]=max(f[i-1][1],f[i-1][0]+prices[i-1]-fee);
}a
return f[prices.size()][1];
}
};