思路:
分为三种情况:
i)不持股,从来没有买进过(0)
ii)持股,买进过一次(1)
iii)不持股,卖掉了一次(2)
设置dp[i][j],i表示下标为i,j表示状态(0~3)
代码:
class Solution {
public int maxProfit(int[] prices) {
int n=prices.length;
int[][] dp=new int[n][3];
if(n<2) return 0;
//dp[i][0]:不持股,从来没有持股过。一直都是0
//dp[i][1]:第一次持股
//dp[i][2]:不持股,已卖出一次
//初始化
dp[0][0]=0;
dp[0][1]=-prices[0];
//状态2不应该有值,则设为0不影响操作
dp[0][2]=0;
//分得这么细是因为只有一次买入和卖出操作,不可能从卖出到买入
for(int i=1;i<n;i++){
//持股:前一个就持股/前一个不持股,现在这个买股
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
//不持股:前一个就不持股/前一个持股,当前麦掉
dp[i][2]=Math.max(dp[i-1][2],dp[i-1][1]+prices[i]);
//可以不显式赋值,因为该状态一直为0,并且int初始值就是0
dp[i][0]=0;
}
return dp[n-1][2];
}
}
分解:
1)是dp的第一种形式,当前值只与前2个结果有关
区间规划
2)dp[i][0]的值一直都是0
不可能有dp[i][2]到dp[i][1]的过程。因为只有一次操作,不可能卖出了之后还有买入的操作
3)这里设置3种情况,其实是把不持股分了两种情况,这样更清晰一点
4)可以不显式赋值,因为该状态一直为0,并且int初始值就是0
dp[i][0]=0;
最后比较不持股,有一次卖出的情况即可,因为持股的情况是负值。不持股,没有买入的情况永远是0