可以用字符串表示一个学生的出勤记录,其中的每个字符用来标记当天的出勤情况(缺勤、迟到、到场)。记录中只含下面三种字符:
'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]