結果
問題 | No.2418 情報通だよ!Nafmoくん |
ユーザー |
|
提出日時 | 2023-08-12 14:36:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 683 ms / 2,000 ms |
コード長 | 2,588 bytes |
コンパイル時間 | 303 ms |
コンパイル使用メモリ | 82,232 KB |
実行使用メモリ | 144,372 KB |
最終ジャッジ日時 | 2024-11-14 12:23:53 |
合計ジャッジ時間 | 8,927 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
import collectionsclass UnionFind():def __init__(self):'''unionfind経路圧縮あり,要素にtupleや文字列可,始めに要素数指定なし'''self.parents = dict() # {子要素:親ID,}self.members_set = collections.defaultdict(lambda : set()) # keyが根でvalueが根に属する要素要素(tupleや文字列可)self.roots_set = set() # 根の集合(tupleや文字列可)self.key_ID = dict() # 各要素にIDを割り振るself.ID_key = dict() # IDから要素名を復元するself.cnt = 0 # IDのカウンターdef dictf(self,x): # 要素名とIDをやり取りするところif x in self.key_ID:return self.key_ID[x]else:self.cnt += 1self.key_ID[x] = self.cntself.parents[x] = self.cntself.ID_key[self.cnt] = xself.members_set[x].add(x)self.roots_set.add(x)return self.key_ID[x]def find(self, x):ID_x = self.dictf(x)if self.parents[x] == ID_x:return xelse:self.parents[x] = self.key_ID[self.find(self.ID_key[self.parents[x]])]return self.ID_key[self.parents[x]]def union(self, x, y):x = self.find(x)y = self.find(y)if self.parents[x] > self.parents[y]:x, y = y, xif x == y:returnfor i in self.members_set[y]:self.members_set[x].add(i)self.members_set[y] = set()self.roots_set.remove(y)self.parents[y] = self.key_ID[x]def size(self, x):#xが含まれる集合の要素数return len(self.members_set[self.find(x)])def same(self, x, y):#同じ集合に属するかの判定return self.find(x) == self.find(y)def members(self, x):#xを含む集合の要素return self.members_set[self.find(x)]def roots(self):#根の要素return self.roots_setdef group_count(self):#根の数return len(self.roots_set)def all_group_members(self):#根とその要素return {r: self.members_set[r] for r in self.roots_set}uf = UnionFind()N, M = map(int, input().split())for i in range(M):a, b = map(int, input().split())uf.union(a, b)ans = Nfor i in uf.roots():ans -= uf.size(i) // 2print(ans)