結果
問題 | No.1112 冥界の音楽 |
ユーザー | xuanji |
提出日時 | 2020-07-10 23:14:32 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 258 ms / 2,000 ms |
コード長 | 1,257 bytes |
コンパイル時間 | 225 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 77,056 KB |
最終ジャッジ日時 | 2024-04-19 23:39:07 |
合計ジャッジ時間 | 3,789 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 59 ms
69,888 KB |
testcase_01 | AC | 39 ms
52,352 KB |
testcase_02 | AC | 40 ms
52,096 KB |
testcase_03 | AC | 39 ms
52,224 KB |
testcase_04 | AC | 39 ms
52,608 KB |
testcase_05 | AC | 39 ms
52,224 KB |
testcase_06 | AC | 70 ms
76,032 KB |
testcase_07 | AC | 39 ms
52,608 KB |
testcase_08 | AC | 44 ms
59,008 KB |
testcase_09 | AC | 63 ms
67,712 KB |
testcase_10 | AC | 39 ms
52,224 KB |
testcase_11 | AC | 60 ms
71,424 KB |
testcase_12 | AC | 38 ms
52,480 KB |
testcase_13 | AC | 64 ms
71,808 KB |
testcase_14 | AC | 108 ms
76,160 KB |
testcase_15 | AC | 77 ms
76,032 KB |
testcase_16 | AC | 39 ms
52,352 KB |
testcase_17 | AC | 79 ms
76,088 KB |
testcase_18 | AC | 79 ms
76,204 KB |
testcase_19 | AC | 40 ms
52,352 KB |
testcase_20 | AC | 71 ms
75,904 KB |
testcase_21 | AC | 110 ms
76,284 KB |
testcase_22 | AC | 109 ms
76,160 KB |
testcase_23 | AC | 75 ms
76,032 KB |
testcase_24 | AC | 40 ms
53,120 KB |
testcase_25 | AC | 40 ms
52,864 KB |
testcase_26 | AC | 61 ms
67,200 KB |
testcase_27 | AC | 62 ms
66,944 KB |
testcase_28 | AC | 72 ms
76,160 KB |
testcase_29 | AC | 40 ms
52,608 KB |
testcase_30 | AC | 125 ms
76,544 KB |
testcase_31 | AC | 72 ms
76,032 KB |
testcase_32 | AC | 125 ms
77,056 KB |
testcase_33 | AC | 60 ms
67,072 KB |
testcase_34 | AC | 39 ms
52,736 KB |
testcase_35 | AC | 60 ms
67,200 KB |
testcase_36 | AC | 258 ms
76,916 KB |
ソースコード
#!/usr/bin/env python3 MODULUS = 10**9+7 class Mat(list): def __matmul__(self, B) : A = self return Mat([[sum(A[i][k]*B[k][j] for k in range(len(B))) % MODULUS for j in range(len(B[0])) ] for i in range(len(A))]) def identity(size): size = range(size) return [[(i==j)*1 for i in size] for j in size] def power(F, n): result = Mat(identity(len(F))) b = Mat(F) while n > 0: if (n%2) == 0: b = b @ b n //= 2 else: result = b @ result b = b @ b n //= 2 return result K, M, N = input().split(' ') K = int(K) M = int(M) N = int(N) adjacency = [] for _ in range(K*K): row = [0]*(K*K) adjacency += [row] for _ in range(M): P, Q, R = input().split(' ') P = int(P) Q = int(Q) R = int(R) P -= 1 Q -= 1 R -= 1 adjacency[P*K + Q][Q*K+R] = 1 if N == 1: print(1) elif N == 2: print(adjacency[0][0]) else: adjacency = power(adjacency, N-2) ret = 0 for p in [0]: for q in range(K): for r in range(K): for s in [0]: ret += adjacency[p*K+q][r*K+s] ret = ret % MODULUS print(ret)