結果

問題 No.1340 おーじ君をさがせ
コンテスト
ユーザー persimmon-persimmon
提出日時 2021-02-12 16:40:53
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 772 bytes
コンパイル時間 165 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 81,792 KB
最終ジャッジ日時 2024-07-19 10:42:35
合計ジャッジ時間 25,072 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

def dot(mtr0,mtr1):
  a,b=len(mtr0),len(mtr1[0])
  t=len(mtr0[0])
  if t!=len(mtr1):return None
  ret=[[0]*b for _ in range(a)]
  for i in range(a):
    for j in range(b):
      tmp=0
      for k in range(t):
        tmp+=mtr0[i][k]*mtr1[k][j]
        if tmp>0:tmp=1
      ret[i][j]=tmp
  return ret

def main0(n,m,t,ab):
  # ダブリング。O(n^2)
  g=[[0]*n for _ in range(n)]
  for a,b in ab:
    a,b=a-1,b-1
    g[b][a]=1
  vc=[[0] for _ in range(n)]
  vc[0][0]=1
  for i in range(60):
    if t&(1<<i):
      vc=dot(g,vc)
    g=dot(g,g)
  ans=0
  for i in range(n):
    if vc[i][0]==1:
      ans+=1
  return ans
  
if __name__=='__main__':
  n,m,t=map(int,input().split())
  ab=[list(map(int,input().split())) for _ in range(m)]
  ret0=main0(n,m,t,ab)
  print(ret0)
0