[ABC 177 D Friends] pythonのロジックエラーについて

Question

前提・実現したいこと

Pythonで下記の問題を解いておりましたが、どうしてもある入力に対して間違いを出力します
(9つの入力例に対して不正解)
どこのロジックが間違っているか、ご教示いただけますでしょうか
問題文リンク

基本的な考え方は、union-findで友人グループを作り、そのうち最大人数を持つグループの人数を出力する、という考えです

該当のソースコード

def main():
    n, m = map(int, input().split())
    par = [i for i in range(n+1)]   
    rank = [0] * (n+1)

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

    for _ in range(m):
        a, b = map(int, input().split())
        unite(a,b)
    
    indic = {}
    for num in par:
        if num not in indic:
            indic[num] = 1
        else:
            indic[num] += 1
    
    print(max(indic.values()))


if __name__ == "__main__":
    main()

試したこと

いくつか思いつくコーナーケースを入力する等してみましたが、ちょっと問題が思い当たりません
恐れ入りますが、ご指摘いただければ幸いです

Answer 1

    indic = {}
    for num in par:
        if num not in indic:
            indic[num] = 1
        else:
            indic[num] += 1

単にparの値を見てもどのグループに属しているかはわかりません。findのロジックを参考にしてください。

Comments

[sasuke_]

ありがとうございました！
仰るように、各グループのサイズの情報を持っていなかったので、そのサイズ情報をリストで持ち、uniteするごとにサイズを増していく、という処理を足すことで解決しました！