結果

問題 No.2316 Freight Train
ユーザー timi
提出日時 2023-05-26 21:27:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,120 ms / 2,000 ms
コード長 720 bytes
コンパイル時間 339 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 114,176 KB
最終ジャッジ日時 2024-12-25 04:52:33
合計ジャッジ時間 25,805 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

N,Q=map(int, input().split())
A=list(map(int, input().split()))

par=[i for i in range(N)]
rank=[0]*(N)
friend=[0]*N
block=[0]*N
size=[1]*N
def find(x):
  if par[x]==x:
    return x
  else:
    par[x]=find(par[x])
    return par[x]
 
#同じ集合か判定
def same(x,y):
  return find(x)==find(y)
 
def union(x,y):
  x=find(x)
  y=find(y)
  if x==y:
    return 
  if rank[x]>rank[y]:
    par[y]=x
    size[x]+=size[y]
  else:
    par[x]=y
    size[y]+=size[x]
    if rank[x]==rank[y]:
      rank[y]+=1
 
 
for a in range(N):
  #a,b=map(int,input().split())
  b=A[a]
  if b!=-1:
    union(a,b-1)
    
for _ in range(Q):
  a,b=map(int,input().split())
  a-=1;b-=1
  if same(a,b):
    print('Yes')
  else:
    print('No')
0