結果

問題 No.2563 色ごとのグループ
ユーザー u_kunu_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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
51,968 KB
testcase_01 AC 39 ms
52,224 KB
testcase_02 AC 43 ms
51,968 KB
testcase_03 AC 45 ms
52,224 KB
testcase_04 AC 40 ms
52,224 KB
testcase_05 AC 39 ms
51,968 KB
testcase_06 AC 40 ms
52,096 KB
testcase_07 AC 40 ms
51,968 KB
testcase_08 AC 40 ms
52,096 KB
testcase_09 AC 43 ms
53,120 KB
testcase_10 AC 39 ms
51,968 KB
testcase_11 AC 43 ms
52,992 KB
testcase_12 AC 42 ms
52,480 KB
testcase_13 AC 41 ms
52,224 KB
testcase_14 AC 81 ms
72,320 KB
testcase_15 AC 82 ms
72,576 KB
testcase_16 AC 82 ms
72,576 KB
testcase_17 AC 80 ms
72,192 KB
testcase_18 AC 57 ms
63,616 KB
testcase_19 AC 110 ms
77,056 KB
testcase_20 AC 124 ms
82,048 KB
testcase_21 AC 105 ms
79,104 KB
testcase_22 AC 115 ms
77,696 KB
testcase_23 AC 103 ms
78,976 KB
testcase_24 AC 261 ms
103,032 KB
testcase_25 AC 220 ms
106,612 KB
testcase_26 AC 315 ms
122,272 KB
testcase_27 AC 317 ms
141,932 KB
testcase_28 AC 326 ms
126,644 KB
testcase_29 AC 384 ms
145,152 KB
testcase_30 AC 409 ms
145,008 KB
testcase_31 AC 398 ms
145,392 KB
testcase_32 AC 413 ms
145,784 KB
testcase_33 AC 442 ms
139,464 KB
testcase_34 AC 450 ms
141,216 KB
testcase_35 AC 463 ms
140,964 KB
testcase_36 AC 458 ms
139,720 KB
testcase_37 AC 454 ms
139,464 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