力扣309-最佳买卖股票时期(含冷冻期)

        买卖股票系列的第三道题,采用的算法也是动态规划算法。核心思想就是计算每天收益时要分三种情况,分别对应当天结束后的交易状态,具体解析过程参考代码示例。

package likou;
/*
 * 最佳买卖股票时期(含冷冻期)
 * 题干:
 * 	给定一个整数数组prices,其中第i个元素代表了第i天的股票价格
 * 	设计一个算法计算出最大利润
 * 	在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票)
 * 	你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
 * 	卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)
 */
public class Demo309 {
	/*
	 * 解题思路:
	 * 	当i=0时,收益肯定是0
	 * 	当i=1时,收益是Math.max(0,prices[1]-prices[0]);
	 *  当i>=2时,就需要分三种情况来看
	 *  	第一种情况就是第i天结束后,手上没有股票且非冷冻期则收益dp0
	 *          第i-1天结束,没有股票非冷冻期或没有股票冷冻期
	 *          对应dp0[i] = Math.max(dp0[i-1],dp1[i-1]);
	 *      第二种情况就是第i天结束,手上没有股票但处于冷冻期则收益dp1
	 *      	股票在第i天卖出且i-1天结束手上有股票
	 *      	对应dp1[i] = dp2[i-1]+prices[i]
	 *      第三种情况就是第i天结束,手上有股票的收益dp2
	 *          第i-1天结束手上有股票或者第i-1天结束手上没有股票且非处于冷冻期
	 *          对应dp2[i] = Math.max(dp2[i-1],dp0[i-1]-prices[i])
	 */
	public int maxProfit(int[] prices) {
          int dp0 = 0;
          int dp1 = 0;
          int dp2 = -prices[0];
          int length = prices.length;
          for(int i=1;i<length;i++) {
        	  int tempdp0 = Math.max(dp0, dp1);
        	  int tempdp1 = dp2+prices[i];
        	  dp2 = Math.max(dp2, dp0-prices[i]);
        	  dp0 = tempdp0;
        	  dp1 = tempdp1;
          }
          return Math.max(dp0, dp1);
    }
	
	public static void main(String args[]) {
		Demo309 demo = new Demo309();
		int[] prices = {1,2,3,0,2};
		System.out.println(demo.maxProfit(prices));
	}
}

上一篇:[NOIP1999]拦截导弹


下一篇:​LeetCode刷题实战276:栅栏涂色