LeetCode--552. 学生出勤记录 II

552. 学生出勤记录 II

思路：动态规划
列出状态转移列表即可，每一天的状态只能由前一天的状态转移过来

/**
 * 552. 学生出勤记录 II
 *
 */

public class Solution552 {
  private static final int MOD = 1_000_000_007;

  public int checkRecord(int n) {
    // 三维dp数组
    // 第一维：当前的天数
    // 第二维：代表以前数组中存在A的个数，A只能有1个或者0个
    // 第三维：代表以前数组中存在连续L的个数，并且连续的L位于结尾处，在结尾处连续的L不能超过2个
    int[][][] dp = new int[n + 1][2][3];
    // dp[i][0][0] 代表结尾是P
    // dp[i][1][0] 代表结尾是A
    // dp[i][0][1] 代表结尾是L
    dp[0][0][0] = 1;
    for (int i = 1; i <= n; i++) {
      // 以P结尾的出勤记录
      for (int j = 0; j <= 1; j++) {
        for (int k = 0; k <=2; k++) {
          // 以P结尾，则连续L的数目会清零，所以当前的状态可以从前一天的任意一个状态转移过来
          dp[i][j][0] = (dp[i][j][0] + dp[i - 1][j][k]) % MOD;
        }
      }

      // 以A结尾的出勤记录
      for (int k = 0; k <= 2; k++) {
        // 以A结尾，连续L的数目会清零，所以当前的状态可以从前一天的P或者L结尾状态转移过来
        dp[i][1][0] = (dp[i][1][0] + dp[i - 1][0][k]) % MOD;
      }

      // 以L结尾的出勤记录
      for (int j = 0; j <= 1; j++) {
        for (int k = 1; k <= 2; k++) {
          // 以L结尾，连续L的数目会增加，所以当前的状态从前一天的P或者A或者L（L数目不能超过1）转移来
          dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j][k - 1]) % MOD;
        }
      }
    }
    int sum = 0;
    for (int j = 0; j <= 1; j++) {
      for (int k = 0; k <= 2; k++) {
        sum = (sum + dp[n][j][k]) % MOD;
      }
    }
    return sum;
  }
}