121. 买卖股票的最佳时机（简单）

思路：

动态规划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）