【LeetCode：552】学生出勤记录 II（Java）

题目链接

• 552. 学生出勤记录 II

题目描述

可以用字符串表示一个学生的出勤记录，其中的每个字符用来标记当天的出勤情况（缺勤、迟到、到场）。记录中只含下面三种字符：
‘A’：Absent，缺勤
‘L’：Late，迟到
‘P’：Present，到场
如果学生能够 同时 满足下面两个条件，则可以获得出勤奖励：

按 总出勤 计，学生缺勤（‘A’）严格 少于两天。
学生 不会 存在 连续 3 天或 连续 3 天以上的迟到（‘L’）记录。
给你一个整数 n ，表示出勤记录的长度（次数）。请你返回记录长度为 n 时，可能获得出勤奖励的记录情况 数量 。答案可能很大，所以返回对 10⁹ + 7 取余 的结果。

示例 1：

输入：n = 2
输出：8
解释：
有 8 种长度为 2 的记录将被视为可奖励：
“PP” , “AP”, “PA”, “LP”, “PL”, “AL”, “LA”, “LL”
只有"AA"不会被视为可奖励，因为缺勤次数为 2 次（需要少于 2 次）。
示例 2：

输入：n = 1
输出：3
示例 3：

输入：n = 10101
输出：183236316

提示：

1 <= n <= 10⁵

求解思路

• 深度优先搜索：cache数组用于缓存每次深搜得到的结果，cache[i][j[k]表示剩余i个数字需要决策时，A出现j次，L出现k次的情况次数。
• public int dfs(int t, int Acnt, int Lcnt)方法中，t表示剩余需要决策的次数，Acnt和Lcnt分别表示缺勤次数和结尾出迟到的连续次数。

实现代码

class Solution {
    final int mod = (int)1e9 + 7;
    int[][][] cache;
    public int checkRecord(int n) {
        cache = new int[n + 1][2][3];
        for (int i = 0; i <= n; i ++) {
            for (int j = 0; j < 2; j ++) {
                for (int k = 0; k < 3; k ++) {
                    cache[i][j][k] = -1;
                }
            }
        }

        return dfs(n, 0, 0);
    }

    public int dfs(int t, int Acnt, int Lcnt) {
        if (Acnt >= 2 || Lcnt >= 3) {
            return 0;
        } else if (t == 0) {
            return 1;
        } else if (cache[t][Acnt][Lcnt] != -1) {
            return cache[t][Acnt][Lcnt];
        }
        int ans = 0;
        ans = dfs(t - 1, Acnt + 1, 0) % mod; //A
        ans = (ans + dfs(t - 1, Acnt, Lcnt + 1)) % mod; //L
        ans = (ans + dfs(t - 1, Acnt, 0)) % mod; //P
        cache[t][Acnt][Lcnt] = ans;
        return ans; 
    }
}