forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution2.cpp
33 lines (29 loc) · 1022 Bytes
/
Solution2.cpp
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
28
29
30
31
32
33
constexpr int MOD = 1e9 + 7;
class Solution {
public:
int checkRecord(int n) {
using ll = long long;
vector<vector<vector<ll>>> dp(n, vector<vector<ll>>(2, vector<ll>(3)));
// base case
dp[0][0][0] = dp[0][0][1] = dp[0][1][0] = 1;
for (int i = 1; i < n; ++i) {
// 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;
}
ll ans = 0;
for (int j = 0; j < 2; ++j) {
for (int k = 0; k < 3; ++k) {
ans = (ans + dp[n - 1][j][k]) % MOD;
}
}
return ans;
}
};