結果

問題 No.2563 色ごとのグループ
ユーザー u_kunu_kun
提出日時 2023-12-02 17:43:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 522 ms / 2,000 ms
コード長 2,520 bytes
コンパイル時間 696 ms
コンパイル使用メモリ 81,700 KB
実行使用メモリ 145,300 KB
最終ジャッジ日時 2023-12-02 17:44:06
合計ジャッジ時間 9,490 ms
ジャッジサーバーID
（参考情報） 		judge9 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
53,460 KB
testcase_01 AC 37 ms
53,460 KB
testcase_02 AC 37 ms
53,460 KB
testcase_03 AC 37 ms
53,460 KB
testcase_04 AC 37 ms
53,460 KB
testcase_05 AC 36 ms
53,460 KB
testcase_06 AC 43 ms
53,460 KB
testcase_07 AC 37 ms
53,460 KB
testcase_08 AC 37 ms
53,460 KB
testcase_09 AC 40 ms
53,460 KB
testcase_10 AC 42 ms
53,460 KB
testcase_11 AC 44 ms
53,460 KB
testcase_12 AC 38 ms
53,460 KB
testcase_13 AC 38 ms
53,460 KB
testcase_14 AC 73 ms
73,004 KB
testcase_15 AC 74 ms
73,004 KB
testcase_16 AC 73 ms
73,020 KB
testcase_17 AC 72 ms
73,020 KB
testcase_18 AC 52 ms
64,452 KB
testcase_19 AC 99 ms
76,716 KB
testcase_20 AC 119 ms
81,896 KB
testcase_21 AC 98 ms
78,916 KB
testcase_22 AC 108 ms
77,528 KB
testcase_23 AC 102 ms
78,404 KB
testcase_24 AC 267 ms
102,552 KB
testcase_25 AC 214 ms
105,476 KB
testcase_26 AC 321 ms
121,796 KB
testcase_27 AC 325 ms
141,452 KB
testcase_28 AC 356 ms
126,568 KB
testcase_29 AC 396 ms
144,792 KB
testcase_30 AC 435 ms
144,788 KB
testcase_31 AC 445 ms
144,788 KB
testcase_32 AC 436 ms
145,300 KB
testcase_33 AC 522 ms
138,988 KB
testcase_34 AC 496 ms
140,752 KB
testcase_35 AC 488 ms
140,232 KB
testcase_36 AC 506 ms
139,116 KB
testcase_37 AC 487 ms
139,112 KB
権限があれば一括ダウンロードができます

ソースコード

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