https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/
枚举型
dp[i][j],前i个,j:0买入,1卖出,冷冻不用管,由于是枚举上一次操作的日期,买入的上一次卖出是前2天之前。
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n=prices.size();
if(n==0)
return 0;
int dp[n][2];
memset(dp,0,sizeof(dp));
for(int i=0;i<n;i++){
dp[i][0]=0;
for(int j=0;j<i;j++){
if(j<i){
dp[i][1]=max(dp[i][1],dp[j][0]+prices[i]-prices[j]);
}
if(j<i-1){
dp[i][0]=max(dp[i][0],dp[j][1]);
}
}
}
int ans=0;
for(int i=0;i<n;i++){
ans=max(ans,dp[i][1]);
}
return ans;
}
};
状态型
dp[i][j],前i个,j:0不持股,1持股,2冷冻期。复杂度O(n),买入的时候减一个数,卖出的时候加一个数,合起来就是一个差。
然后状态是完成动作后的状态,这样是因为对于未来的读取过去转移更方便。
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n=prices.size();
if(n==0)
return 0;
int dp[n][3];
memset(dp,0,sizeof(dp));
dp[0][1]=-prices[0];
for(int i=1;i<n;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i]);
dp[i][1]=max(dp[i-1][1],dp[i-1][2]-prices[i]);
dp[i][2]=dp[i-1][0];
}
return max(dp[n-1][0],dp[n-1][2]);
}
};