結果

問題 No.1340 おーじ君をさがせ
コンテスト
ユーザー Navier_Boltzmann
提出日時 2022-03-10 10:47:08
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,710 bytes
コンパイル時間 420 ms
コンパイル使用メモリ 82,180 KB
実行使用メモリ 87,472 KB
最終ジャッジ日時 2024-09-14 11:51:40
合計ジャッジ時間 4,841 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 1 -- * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
import sys
input = sys.stdin.buffer.readline
from functools import reduce
##fの単位元を要考慮、matrix_powのCの初期化

def f(x,y):
    return x|y
def g(x,y):
    return x&y
def matrix_mul(A,B,f,g,mod = None):
    nA = len(A)
    mA = len(A[0])
    mB = len(B[0])
    
    tmp = [[0]*mB for _ in range(nA)]
    
    if mod is None:
        for i in range(nA):
            for j in range(mB):
                tmp[i][j] = reduce(f,(g(A[i][k],B[k][j]) for k in range(mA)))
                
        return tmp
    
    for i in range(nA):
        for j in range(mB):
            tmp[i][j] = reduce(f,(g(A[i][k],B[k][j])%mod for k in range(mA)))%mod
    return tmp
    

def matrix_pow(A,n,f,g,mod = None):
    nbit = list(str(bin(n))[2:])
    nbit = [int(i) for i in nbit]
    N = len(A)
    C = [[False]*N for _ in range(N)]
    B = A
    for i in range(N):
        C[i][i] = True
        
    if mod is None:
        if f is None:
            for i in range(len(nbit)):
                if nbit[-1-i] == 1:
                    C = matrix_mul(C,B,f,g)
                
                B = matrix_mul(B,B,f,g)
            return C
        
    for i in range(len(nbit)):
        if nbit[-1-i] == 1:
            C = matrix_mul(C,B,f,g,mod=mod)
        
        B = matrix_mul(B,B,f,g,mod=mod)
    
    return C
    
N,M,T = map(int,input().split())
A = [[False]*N for _ in range(N)]

for _ in range(M):
    a,b = map(int,input().split())
    A[b][a] = True

INIT = [[False] for _ in range(N)]
INIT[0] = [True]

X = matrix_pow(A,T,f,g)
Y = matrix_mul(X,INIT,f,g)
ans = sum(int(*Y[i]) for i in range(N))
#print(*Y[0])
print(ans)
0