結果
問題 | No.2563 色ごとのグループ |
ユーザー | Cecil |
提出日時 | 2023-12-02 16:09:20 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,909 bytes |
コンパイル時間 | 932 ms |
コンパイル使用メモリ | 12,160 KB |
実行使用メモリ | 47,332 KB |
最終ジャッジ日時 | 2023-12-02 16:09:39 |
合計ジャッジ時間 | 16,234 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge10 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 30 ms
10,240 KB |
testcase_01 | AC | 30 ms
10,240 KB |
testcase_02 | AC | 30 ms
10,240 KB |
testcase_03 | AC | 30 ms
10,240 KB |
testcase_04 | AC | 30 ms
10,240 KB |
testcase_05 | AC | 30 ms
10,240 KB |
testcase_06 | AC | 30 ms
10,240 KB |
testcase_07 | AC | 30 ms
10,240 KB |
testcase_08 | AC | 30 ms
10,240 KB |
testcase_09 | AC | 31 ms
10,240 KB |
testcase_10 | AC | 31 ms
10,240 KB |
testcase_11 | AC | 32 ms
10,240 KB |
testcase_12 | AC | 31 ms
10,240 KB |
testcase_13 | AC | 31 ms
10,240 KB |
testcase_14 | AC | 37 ms
10,496 KB |
testcase_15 | AC | 39 ms
10,496 KB |
testcase_16 | AC | 38 ms
10,496 KB |
testcase_17 | AC | 38 ms
10,496 KB |
testcase_18 | AC | 34 ms
10,624 KB |
testcase_19 | AC | 54 ms
10,496 KB |
testcase_20 | AC | 105 ms
13,440 KB |
testcase_21 | AC | 74 ms
11,776 KB |
testcase_22 | AC | 66 ms
11,264 KB |
testcase_23 | AC | 95 ms
11,520 KB |
testcase_24 | AC | 716 ms
25,780 KB |
testcase_25 | AC | 597 ms
27,900 KB |
testcase_26 | AC | 884 ms
34,864 KB |
testcase_27 | AC | 848 ms
46,920 KB |
testcase_28 | AC | 990 ms
41,124 KB |
testcase_29 | AC | 1,325 ms
47,320 KB |
testcase_30 | AC | 1,326 ms
47,312 KB |
testcase_31 | AC | 1,335 ms
47,324 KB |
testcase_32 | AC | 1,344 ms
47,332 KB |
testcase_33 | TLE | - |
testcase_34 | -- | - |
testcase_35 | -- | - |
testcase_36 | -- | - |
testcase_37 | -- | - |
ソースコード
from collections import defaultdict as dd class UnionFind(): #初期化 def __init__(self, n): self.par = [-1] * n self.rank = [0] * n self.siz = [1] * n #根を求める def root(self, x): if self.par[x] == -1: return x#xが根の場合はxを返す else: self.par[x] = self.root(self.par[x]) #経路圧縮 return self.par[x] #xとyが同じグループに属するか(根が一致するか) def issame(self, x, y): return self.root(x) == self.root(y) #xを含むグループとyを含むグループを併合する def unite(self, x, y): #x側とy側の根を取得する rx = self.root(x) ry = self.root(y) if rx == ry: return False #すでに同じグループの時は何もしない #union by rank if self.rank[rx] < self.rank[ry]: #ry側のrankが小さくなるようにする rx, ry = ry, rx self.par[ry] = rx #ryをrxの子とする if self.rank[rx] == self.rank[ry]:#ry側のrankを調整する self.rank[rx] += 1 self.rank[rx] += self.siz[ry] #rx側のsizを調整する return True #xを含む根付き木のサイズを求める def size(self, x): return self.siz[self.root(x)] N,M = map(int, input().split()) C = list(map(int, input().split())) uf = UnionFind(N+1) col = dd(list) for i in range(N): col[C[i]].append(i+1) for _ in range(M): u,v = map(int,input().split()) if C[u-1] != C[v-1]: continue uf.unite(u,v) ans = 0 for key in col.keys(): n = len(col[key]) for i in range(n): for j in range(i, n): if i == j: continue if not uf.issame(col[key][i],col[key][j]): uf.unite(col[key][i],col[key][j]) ans += 1 print(ans)