309. El mejor momento para comprar y vender acciones incluye un período de congelación (medio)

Ideas:

Dividido en tres situaciones:

i) Participación (dp [1])

ii) No tenencia de acciones, no en período de congelación (dp [0])

iii) No tenencia de acciones, en período de congelación (dp [2])

 

Código:

class Solution {
    public int maxProfit(int[] prices) {
		int n=prices.length;
		if(n<2) return 0;
		int[][] dp=new int[n][3];
		
		//初始化
		//0:不持股，不在冰冻期
		//1:持股
		//2：不持股，在冰冻期
		dp[0][0]=0;
		dp[0][1]=-prices[0];
		dp[0][2]=0;
		
		//股票卖掉后进入冰冻期
		for(int i=1;i<n;i++){
			//不在冰冻期：前一个不在冰冻期/前一个就在冰冻期，现在这个变为不冰冻期
			dp[i][0]=Math.max(dp[i-1][0],dp[i-1][2]);
			//持股：前一个就持股/前一个不持股并且不在冰冻期，现在买入了股票
			dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
			//不持股在冰冻期：前一个持股卖掉了。不可能前一个还是冰冻期，冰冻期只有1天
			dp[i][2]=dp[i-1][1]+prices[i];
			
		}
		//比较不持股：在冰冻期/不在冰冻期，哪一个更多钱。不比较还持股的
		return Math.max(dp[n-1][0],dp[n-1][2]);
	}
}

 

descomponer:

1) Aquí dp es la primera forma, y ​​el valor actual solo está relacionado con el resultado anterior

Planificación de intervalos

El conjunto dp matriz bidimensional, dp [i] [j] representa la situación de beneficio en el intervalo [0, i] y el estado es j

j tiene 3 situaciones:

    i) Entrepierna (1)

    ii) No tenencia de acciones, no en período de congelación (0)

    iii) No tenencia de acciones, en período de congelación (2)

 

2) Hay un punto a tener en cuenta: después de vender las acciones, debe haber un período de congelación de un día.

//不在冰冻期：前一个不在冰冻期/前一个就在冰冻期，现在这个变为不冰冻期
dp[i][0]=Math.max(dp[i-1][0],dp[i-1][2]);
//持股：前一个就持股/前一个不持股并且不在冰冻期，现在买入了股票
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
//不持股在冰冻期：前一个持股卖掉了。不可能前一个还是冰冻期，冰冻期只有1天
dp[i][2]=dp[i-1][1]+prices[i];

 

3) Finalmente, compare el valor de beneficio final de las dos situaciones:

    i) No tenencia de acciones + no en período de congelación

    ii) No tenencia de acciones + en el período de congelación

No compare la situación de tener todavía existencias, porque todavía hay existencias disponibles, no puede ser el máximo