結果
問題 | No.2316 Freight Train |
ユーザー |
![]() |
提出日時 | 2023-05-26 21:38:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 879 ms / 2,000 ms |
コード長 | 1,480 bytes |
コンパイル時間 | 211 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 110,080 KB |
最終ジャッジ日時 | 2024-12-25 05:39:20 |
合計ジャッジ時間 | 19,901 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
from collections import defaultdictclass UnionFind():def __init__(self, n):self.n = nself.root = [-1]*(n+1)self.rank = [0]*(n+1)def find(self, x):if(self.root[x] < 0):return xelse:self.root[x] = self.find(self.root[x])return self.root[x]def unite(self, x, y):x = self.find(x)y = self.find(y)if(x == y):returnelif(self.rank[x] > self.rank[y]):self.root[x] += self.root[y]self.root[y] = xelse:self.root[y] += self.root[x]self.root[x] = yif(self.rank[x] == self.rank[y]):self.rank[y] += 1def same(self, x, y):return self.find(x) == self.find(y)def size(self, x):return -self.root[self.find(x)]def roots(self):return [i for i, x in enumerate(self.root) if x < 0]def group_size(self):return len(self.roots())def group_members(self):group_members = defaultdict(list)for member in range(self.n):group_members[self.find(member)].append(member)return group_membersN,Q=map(int,input().split())P=list(map(int,input().split()))uf=UnionFind(N)for i in range(N):if P[i]!=-1:uf.unite(i+1,P[i])for i in range(Q):a,b=map(int,input().split())if uf.same(a,b):print('Yes')else:print('No')