【Leetcode】552. Student Attendance Record II

题目地址：

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

给定一个学生$n$天的出勤记录，'A'表示缺席，'L'表示迟到，'P'表示出席。要求缺席总数不能超过或等于$2$，并且不能有连续$3$或更多的天数迟到。问他要满足要求，总共有多少种出席方式。答案模$10^9+7$返回。

设$f[i][j][k]$是一共$i$天，总缺席数是$j$并且以$k$个迟到结尾的情况下，满足条件的出席方式数。那么考虑第$i+1$天，如果其是出席，那么可以贡献方案数给$f[i+1][j][0]$；如果其是缺席，那么可以贡献方案数给$f[i+1][j+1][0]$（在$j+1<2$的情况下，否则不合法了）；如果其是迟到，那么可以贡献方案数给$f[i+1][j][k+1]$（在$k+1<3$的情况下，否则不合法）。初始$f[0][0][0]=1$。代码如下：

public class Solution {
    public int checkRecord(int n) {
        int MOD = (int) (1e9 + 7);
        int[][][] f = new int[n + 1][2][3];
        f[0][0][0] = 1;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < 2; j++) {
                for (int k = 0; k < 3; k++) {
                    if (j + 1 < 2) {
                        f[i + 1][j + 1][0] = (f[i + 1][j + 1][0] + f[i][j][k]) % MOD;
                    }
                    f[i + 1][j][0] = (f[i + 1][j][0] + f[i][j][k]) % MOD;
                    if (k + 1 < 3) {
                        f[i + 1][j][k + 1] = (f[i + 1][j][k + 1] + f[i][j][k]) % MOD;
                    }
                }
            }
        }
        
        int res = 0;
        for (int j = 0; j < 2; j++) {
            for (int k = 0; k < 3; k++) {
                res = (res + f[n][j][k]) % MOD;
            }
        }
        
        return res;
    }
}

时空复杂度$O(n)$。