LeetCode.552 学生出勤记录 II

原题

https://leetcode-cn.com/problems/student-attendance-record-ii/

思路

递归回溯
递归回溯的多样化
递归回溯的记忆化

给一个大佬的链接，我只研究了方法一和方法二

题解

package com.leetcode.code;

public class Code552 {

    public static void main(String[] args) {
        System.out.println(checkRecord(5));
    }

    static int MOD = 1000000007;

    public static int checkRecord(int n) {
        int[][][] memo = new int[n][2][3];
        return dfs(0, n, 0, 0, memo);
    }

    private static int dfs (int day, int n, int absent, int late, int[][][] memo) {
        if (day >= n) {
            return 1;
        }

//        遇到相同状态的记忆结果直接return
        if (memo[day][absent][late] != 0) {
            return memo[day][absent][late];
        }

//        开始递归回溯
        int res = 0;

//        "P" 随便放置
//        参数说明：absent由于A不要求连续，所以下次递归时要求保证A的数量
//        参数说明：late由于要求连续不能超过2个，所以有L时递归到此时，为放置P的时刻，所以late的数量重新置为0
        res = (res + dfs(day+1, n, absent, 0, memo)) % MOD;

//        "A" 最多只能放置1个
        if (absent < 1) {
            res = (res + dfs(day+1, n, 1, 0, memo)) % MOD;
        }

//        "L" 最多连续放置2个
        if (late < 2) {
            res = (res + dfs(day+1, n, absent, late+1, memo)) % MOD;
        }

        memo[day][absent][late] = res;
        return res;
    }

}