在我刚开始使用并查集这种算法的时候,我一直搞不明白为什么别人总是说这种算法很好,在做题的时候会经常用到。但是我自己用起来的时候时间总是会超,一直很纳闷。今天重新做到一道别人都是使用并查集来解决的题目时,我才发现了我一直以来的错误!是真蠢呀!写这篇博客来吐槽一下自己,顺便给自己留下一个更深刻的印象。
问题就出现在我的find_root函数(查找节点的根节点,也就是查看节点属于哪颗树)上。直接看代码吧。
//错误版
int find_root(int pos)
{
if (forest[pos] == -1)
return pos;
return find_root(forest[pos]);
}
//正确版
int find_root(int pos)
{
if (forest[pos] == -1)
return pos;
return forest[pos] = find_root(forest[pos]);
}