結果
問題 | 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 collections class 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 += 1 self.key_ID[x] = self.cnt self.parents[x] = self.cnt self.ID_key[self.cnt] = x self.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 x else: 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, x if x == y: return for 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_set def 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 = N for i in uf.roots(): ans -= uf.size(i) // 2 print(ans)