AtCoder / 友達の友達

ふくほです。
ABC016-C 友達の友達
を解きました。
atcoder.jp

考えたこと

友達の友達の友達の…
だとUnionFindで管理することが
最適ですが、
今回は友達の友達までなので
木は使いません。

ユーザーの友達を2重vectorで
管理します。
ユーザーの友達の一人を
としたとき、の友達を全探索します。
の友達をとしたとき
とは友達でないことと
まだが友達の友達として
数えられていないこと
の2つを確認します。
3重ループになりますが、
制約が小さいので大丈夫です。

提出コード

atcoder.jp

感想

はじめUnionFindで実装してて
あれあれ？ってなってました。
bool配列最高。
vectorの要素で回すfor文、
簡潔で分かりやすくて好き。