結果

問題 No.2563 色ごとのグループ
ユーザー ありあけ
提出日時 2024-09-28 14:45:30
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,009 ms / 2,000 ms
コード長 1,483 bytes
コンパイル時間 215 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 35,644 KB
最終ジャッジ日時 2024-09-28 14:45:48
合計ジャッジ時間 12,948 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict
N,M = map(int,input().split())
C = list(map(int,input().split()))
if len(set(C)) == len(C):
    print(0)
    exit()


par = [i for i in range(N+1)]
#parはparent親のこと
#par[x]の値がその木の根=親を表す
#print(par)
def find(x):
    if par[x] == x:
        return x
    else:
        par[x] = find(par[x])
        #ここ経路圧縮 テクニックですよねえ
        return par[x]

def same(x,y):
    return find(x) == find(y)

def same_print(x,y):
    if find(x) == find(y):
        print(x)
    else:
        return False
    #これ微妙か?intが返されるか
    # Falseが返されるかわかんない上に
    # 0だとFalseになるの最悪かも

def unite(x,y):
    x = find(x)
    y = find(y)
    if x == y:
        return 0
    if par[x] < par[y]:
        x,y = y,x
    par[x] = y

def members(x):
    root = find(x)
    return [i for i in range(N) if find(i) == root]

def roots():
    return [i for i,x in enumerate(par) if x == i]

def group_count():
    k = roots()
    return len(k)

def all_group_menbers():
    group_members = defaultdict(list)
    for member in range(N):
        group_members[find(member)].append(member)
    return group_members

for m in range(M):
    u,v = map(int,input().split())
    if C[u-1] == C[v-1]:
        unite(u,v)

print(group_count()-1 - len(set(C)))
#print(graph)
#print(C)
#print(set(C))
#print(len(set(C)))
#print(par)
#print(set(par))
#print(len(set(par))-1)
0