結果

問題 No.2563 色ごとのグループ
ユーザー u_kun
提出日時 2023-12-02 17:43:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 463 ms / 2,000 ms
コード長 2,520 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 145,784 KB
最終ジャッジ日時 2024-09-26 21:17:52
合計ジャッジ時間 8,489 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

# Union-Find 木
class unionfind:
    # n 頂点の Union-Find 木を作成
    # ここでは頂点番号が 1-indexed になるように実装している(先頭はダミー)が、
    # 0-indexed の場合は par, size の長さは n でよい
    # par[i] := 頂点iの1個上の親の頂点番号
    # size[i] := 頂点番号iを最上位(根)としたときの, そのグループに属する頂点の数
    def __init__(self, n):
        self.n = n
        self.par = [ -1 ] * (n + 1) # 最初は親が無い
        self.size = [ 1 ] * (n + 1) # 最初は各グループ(個別にばらばら)の頂点数が 1
    
    # 頂点 x の根(最上位)を返す関数
    def root(self, x):
        # 1 個上（親）がなくなる（つまり根に到達する）まで、1 個上（親）に進み続ける
        while self.par[x] != -1:
            x = self.par[x]
        return x
    
    # 要素 u, v を統合する関数
    def unite(self, u, v):
        rootu = self.root(u)
        rootv = self.root(v)
        # u と v が異なるグループのときのみ処理を行う
        if rootu != rootv:
            # Union by size (グループ内に属する頂点が多い方が根になるように統合する)
            if self.size[rootu] < self.size[rootv]:
                self.par[rootu] = rootv
                self.size[rootv] += self.size[rootu]
            else:
                self.par[rootv] = rootu
                self.size[rootu] += self.size[rootv]
    
    #  要素 u と v が同一のグループかどうかを返す関数
    def same(self, u, v):
        return self.root(u) == self.root(v)


N, M = map(int, input().split())
C = list(map(int, input().split()))

Trans = [None]

#色iの頂点数はcnt[i]個
cnt = [0 for _ in range(N + 1)]
for i in range(N):
    cnt[C[i]] += 1
    
    Trans.append(cnt[C[i]])

#print(cnt)
#print(Trans)

# UF_listのi番目は, 色iのグラフのuf木
UF_list = []

for i in range(N + 1):

    uf = unionfind(cnt[i])
    UF_list.append(uf)



for _ in range(M):
    a, b = map(int, input().split())
    if C[a - 1] != C[b - 1]:
        pass
    else:
        color = C[a - 1]
        #print(a, b, color)
        uf = UF_list[color]
        uf.unite(Trans[a], Trans[b])

ans = 0
done = set()
for i in range(N + 1):
    if cnt[C[i - 1]] >= 2 and C[i - 1] not in done:
        uf = UF_list[C[i - 1]]
        ans += uf.par.count(-1) - 2
        
        #print(C[i - 1], uf.par)
        
        done.add(C[i - 1])

#print()
print(ans)
0