309. 最佳买卖股票时机含冷冻期

image.png

代码

  1. class Solution {
  2.     public int maxProfit(int[] p) {
  4.         /**
  5.          * 1.确定dp数组以及下标含义
  6.          * dp[i][j],第i天状态为j，所生的最多现金为dp[i][j]
  7.          * j -> 0 状态一：买入股票状态（今天买入股票，或者是之前就买入了股票然后没有操作）
  8.          * 卖出股票状态
  9.          *   1） j -> 1 状态二： 两天前就卖出了股票，度过了冷却期，一直没操作，今天保持卖出股票状态
  10.          *   2） j -> 2 状态三： 今天就卖出了股票
  11.          * j -> 3 状态四： 今天为冷却器，但冷却状态不可持续。只有一天！
  12.          */
  14.         if( p == null || p.length < 2) return 0;
  16.         int size = p.length;
  18.         int [][] dp = new int [size][4];
  20.         dp[0][0] = -p[0]; //持有股票
  22.         for(int i = 1; i < size; i++ ) {
  23.             dp[i][0] = Math.max(dp[i - 1][0],Math.max(dp[i - 1][1], dp[i - 1][3]) - p[i]);
  25.             dp[i][1] = Math.max(dp[i - 1][3],dp[i - 1][1]);
  27.             dp[i][2] = dp[i - 1][0] + p[i];
  29.             dp[i][3] = dp[i - 1][2];
  30.         }
  32.         return Math.max(Math.max(dp[size - 1][1],dp[size - 1][2]), dp[size -1][3]);
  34.     }
  35. }