原题保证图连通（本题 assert 后也是如此），这是本题一个较为重要的性质。

另外建议升紫色。