-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path12850.py
47 lines (33 loc) · 795 Bytes
/
12850.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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# 본대 산책2
import sys
input = sys.stdin.readline
MOD = 1000000007
def multiply(a, b):
result = []
for i in range(8):
line = []
for j in range(8):
line.append(sum(a[i][k] * b[k][j] % MOD for k in range(8)) % MOD)
result.append(line)
return result
roads = [
[0, 1, 1, 0, 0, 0, 0, 0],
[1, 0, 1, 1, 0, 0, 0, 0],
[1, 1, 0, 1, 1, 0, 0, 0],
[0, 1, 1, 0, 1, 1, 0, 0],
[0, 0, 1, 1, 0, 1, 1, 0],
[0, 0, 0, 1, 1, 0, 0, 1],
[0, 0, 0, 0, 1, 0, 0, 1],
[0, 0, 0, 0, 0, 1, 1, 0],
]
dp = [[0] * 8 for _ in range(8)]
for i in range(8):
dp[i][i] = 1
D = int(input())
while D > 0:
if D % 2 != 0:
dp = multiply(dp, roads)
D -= 1
roads = multiply(roads, roads)
D //= 2
print(dp[0][0])