思路:
动态规划dp
代码:
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)分为3种状态:
dp[i][0]:不持股,从来没有持股过。一直都是0
dp[i][1]:第一次持股
dp[i][2]:不持股,已卖出一次
初始化:
dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0; 状态2不应该有值,则设为0不影响操作
2)核心代码:
分得这么细是因为只有一次买入和卖出操作,不可能从卖出到买入
//持股:前一个就持股/前一个不持股,现在这个买股
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;
持股的情况:
i)前一个就持股
ii)前一个不持股,现在这个买股
不持股的情况:(第二种不持股的情况:卖掉后的不持股)
i)前一个就不持股
ii)前一个持股,当前卖掉
一直不持股的情况:一直是0
复杂度分析:
时间复杂度:O(N)
空间复杂度:O(N)