結果
問題 | No.2316 Freight Train |
ユーザー |
![]() |
提出日時 | 2023-05-26 21:28:35 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 948 ms / 2,000 ms |
コード長 | 1,039 bytes |
コンパイル時間 | 348 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 109,440 KB |
最終ジャッジ日時 | 2024-12-25 04:55:34 |
合計ジャッジ時間 | 22,707 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
#yukicoder390C #UnionFind class UnionFind: def __init__(self,N): self._parent=[-1 for i in[0]*N] def find(self,v): #頂点vの親を探し、経路圧縮する vertices=[] while self._parent[v]>=0: vertices.append(v);v=self._parent[v] for i in vertices: self._parent[i]=v return v def unite(self,x,y): #頂点xとyを併合し、併合の有無を返す x,y = self.find(x),self.find(y) if x==y: return 0 if self._parent[x]>self._parent[y]: x,y=y,x #負値で管理 self._parent[x]+=self._parent[y]; self._parent[y]=x; return 1 def same(self,x,y):return self.find(x)==self.find(y) #xとyは同一集合か返す def size(self,x): return -self._parent[self.find(x)] #xの集合のサイズを求める f=lambda:list(map(int,input().split())) N,Q=f(); P=f(); UF=UnionFind(N) for i in range(N): x=P[i] if x==-1: continue x-=1 UF.unite(i,x) for _ in range(Q): a,b=f(); a-=1; b-=1 print('Yes') if UF.same(a,b) else print('No')