定义dp[n][k][m],表示第n天结束后,状态为m,已经购买的股票数为k。m=1表示持有股票,m=0表示当前不持有股票。k的取值为0,1,2。写的时候迷迷糊糊的,感觉还是没有掌握精髓。。。。
代码:
class Solution {
public:
int dp[100000][3][2];
int min_int = -2000000000; //不存你在的状态设置为很小,因为是求最大值
int maxProfit(vector<int>& prices) {
int size=prices.size();
if(size==0)
return 0;
dp[0][0][1]=min_int;
dp[0][0][0]=0;
dp[0][1][1]=-prices[0];
dp[0][1][0]=min_int;
dp[0][2][0]=min_int;
dp[0][2][1]=min_int;
for(int i=1;i<size;i++){
for(int k=0;k<3;k++){
dp[i][k][0]=max(dp[i-1][k][0],dp[i-1][k][1]+prices[i]);
if(k-1>=0)
dp[i][k][1]=max(dp[i-1][k][1],dp[i-1][k-1][0]-prices[i]);
else
dp[i][k][1]=min_int;
}
}
return max(dp[size-1][0][0],max(dp[size-1][1][0],dp[size-1][2][0]));
}
};