forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution2.py
27 lines (24 loc) · 909 Bytes
/
Solution2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution:
def checkRecord(self, n: int) -> int:
mod = int(1e9 + 7)
dp = [[[0, 0, 0], [0, 0, 0]] for _ in range(n)]
# base case
dp[0][0][0] = dp[0][0][1] = dp[0][1][0] = 1
for i in range(1, n):
# A
dp[i][1][0] = (dp[i - 1][0][0] + dp[i - 1][0][1] + dp[i - 1][0][2]) % mod
# L
dp[i][0][1] = dp[i - 1][0][0]
dp[i][0][2] = dp[i - 1][0][1]
dp[i][1][1] = dp[i - 1][1][0]
dp[i][1][2] = dp[i - 1][1][1]
# P
dp[i][0][0] = (dp[i - 1][0][0] + dp[i - 1][0][1] + dp[i - 1][0][2]) % mod
dp[i][1][0] = (
dp[i][1][0] + dp[i - 1][1][0] + dp[i - 1][1][1] + dp[i - 1][1][2]
) % mod
ans = 0
for j in range(2):
for k in range(3):
ans = (ans + dp[n - 1][j][k]) % mod
return ans