AtCoderの問題(ABCのE問題)でWAになる

Question

発生している問題

AtCoderの問題が解けません。
https://atcoder.jp/contests/abc126/tasks/abc126_e
この問題で、以下のアルゴリズムを試したところ、一部のテストケースでWAになりました。
他の人のソースコードを見ても同じようなプログラムを組んでいる人が見つからないので、WAの理由が分かりません。教えていただければ幸いです。
（プログラミング初学者なので、簡単なミスだったりするかもしれません。その際はお目汚し失礼しました。）

採用したアルゴリズム

入力されたX,Yをリストx,yにそれぞれ入れる。
x[i]とy[i]は、一方の値が分かればもう一方も自動的に決定できる。
辞書cを用意して、x[i]が辞書のkeyとして存在していれば、y[i]というkeyにc[x[i]]を格納する。その逆の場合の処理も行う。もしx[i]もy[i]も辞書の中になければ、新たな値をx[i],y[i]をkeyとして格納する。こうして、格納されている値が0であるようなkey達はそのうち一つが分かればすべて判明し、格納されている値が1であるようなkey達はそのうち一つが分かればすべて判明し…という性質をもった辞書が作られる。
この辞書のkeyと値をそれぞれリストにする。それらをset型にして、重複をなくす。値のほうのsetのサイズは、「このサイズ分のコストを使えばx,yとして与えられた番目の数は全て確定できる」という意味を持っている。これに、「x,yとして与えられることのなかったノーヒントの番目のカードを魔法で確認する」という操作を行わなければいけない回数を足す。それが期待する出力である。

ソースコード

下手だと思いますが許してください…。

n, m = map(int, input().split())
x = [0 for i in range(m)]
y = [0 for i in range(m)]
z = [0 for i in range(m)]
for i in range(m):
    x[i], y[i], z[i] = map(int, input().split())
c = {}
sorting = 0
for i in range(m):
    if x[i] in c:
        c[y[i]] = c[x[i]]
    elif y[i] in c:
        c[x[i]] = c[y[i]]
    else:
        c[x[i]] = sorting
        c[y[i]] = sorting
        sorting += 1
keylist = []
wordlist = []
for key, word in c.items():
    keylist.append(key)
    wordlist.append(word)
setword = set(wordlist)
setn = set(range(1,n+1))
notn = setn - set(keylist)
print(len(setword) + len(notn))

Answer 1

例えば、1と2の組、3と4の組と来た後、2と3の組が来た場合、
1と2がグループ0、3と4がグループ1となり、その後3がグループ0に組みなおされ、4がグループ1に取り残されます。本来であれば4もグループ0に入り、1から4のどれがわかっても全部わかるという状況のはずです。
なので、値割り当ての際に、相手方に値が割り当てられていないかもチェックし、あった場合はそのグループ全部を書き換えないといけないと思います。

Comments

[lqa]

ありがとうございます！
完全に盲点でした、助かりました。