結果
問題 | No.1340 おーじ君をさがせ |
ユーザー |
👑 ![]() |
提出日時 | 2021-01-15 22:08:32 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 705 ms / 2,000 ms |
コード長 | 1,160 bytes |
コンパイル時間 | 1,150 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 80,640 KB |
最終ジャッジ日時 | 2024-11-27 23:10:13 |
合計ジャッジ時間 | 11,234 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 59 |
ソースコード
"""NFAの状態数行列べき乗じゃん!!"""import sysfrom sys import stdinmod = 998244353#行列A,Bの積(不可能だとエラー,modに0以下の値を指定するとmodを取らなくなる)def matrix_mul(A,B):ans = [ [0] * len(B[0]) for i in range(len(A)) ]for ai in range(len(A)):for bj in range(len(B[0])):now = 0for same in range(len(A[0])):now |= A[ai][same] & B[same][bj]ans[ai][bj] = nowreturn ans#行列Aのx乗(当然正方行列じゃないとだめ)def matrix_pow(A,x):B = [[A[i][j] for j in range(len(A[0]))] for i in range(len(A))]ans = [[0] * len(A[0]) for i in range(len(A))]for i in range(len(A)):ans[i][i] = 1while x > 0:if x % 2 == 1:ans = matrix_mul(ans,B)B = matrix_mul(B,B)x//=2return ansN,M,T = map(int,stdin.readline().split())H = [[0] * N for i in range(N)]for i in range(M):a,b = map(int,stdin.readline().split())H[a][b] = 1A = [[0] * N]A[0][0] = 1ans = matrix_mul(A,matrix_pow(H,T))#print (ans)print (sum(ans[0]))