LeetCode-Python-552. 学生出勤记录 II(DFS Memo + DP)

可以用字符串表示一个学生的出勤记录，其中的每个字符用来标记当天的出勤情况（缺勤、迟到、到场）。记录中只含下面三种字符：

• 'A'：Absent，缺勤
• 'L'：Late，迟到
• 'P'：Present，到场

如果学生能够 同时 满足下面两个条件，则可以获得出勤奖励：

• 按 总出勤 计，学生缺勤（'A'）严格 少于两天。
• 学生 不会 存在 连续 3 天或 连续 3 天以上的迟到（'L'）记录。

给你一个整数 n ，表示出勤记录的长度（次数）。请你返回记录长度为 n 时，可能获得出勤奖励的记录情况 数量 。答案可能很大，所以返回对 109 + 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 <= 105

第一种思路：

我们可以从每个位置上填什么的角度进行 DFS 搜索，

P是一定可以填的， A和L要根据之前填了什么来决定。

这种思路是正确的，但是会爆内存，所以想要 AC 需要把 DFS 函数放在外侧进行全局测试用例共享 cache。

时间复杂度：O（N）

空间复杂度：O（N）

MOD = 10 ** 9 + 7
@cache
def dfs(day, A_cnt, L_consective_cnt):
    if day == 0:
        return 1
    # P
    res = dfs(day - 1, A_cnt, 0)
    # A
    if A_cnt < 1:
        res += dfs(day - 1, A_cnt + 1, 0)
    # L
    if L_consective_cnt < 2:
        res += dfs(day - 1, A_cnt, L_consective_cnt + 1)
    return res % MOD
    
class Solution:
    def checkRecord(self, n: int) -> int:
        return dfs(n, 0, 0)

第二种思路：

DFS + memo 都可以 1：1 翻译成 DP。

L 只有三种可能（0， 1， 2）， A 只有两种可能（0， 1），

根据刚才的返回值 = dfs(n, 0, 0)，所以现在的返回值是 DP[n][0][0]。

因此可以得到 DP 方程的维度是(n + 1)(2)(3)，注意顺序和 DFS 的参数顺序相反。

由 DFS 的 base 可知，DP[0] 对应的应该是 [[1, 1,1], [1, 1, 1]]，即全为 1 的情况。

剩下的就是翻译即可。

时间复杂度：O（N）

空间复杂度：O（N）

class Solution:
    def checkRecord(self, n: int) -> int:
        # return dfs(n, 0, 0)
        dp = [[[0, 0, 0] for x in range(2)] for _ in range(n + 1)]
        dp[0] = [[1, 1, 1], [1, 1, 1]]
        for i in range(1, n + 1):
            for a in range(2):
                for l in range(3):
                    dp[i][a][l] = dp[i - 1][a][0]
                    if a < 1:
                        dp[i][a][l] += dp[i - 1][a + 1][0]
                    if l < 2:
                        dp[i][a][l] += dp[i - 1][a][l + 1]
                    dp[i][a][l] %= MOD
        return dp[n][0][0]