关于并查集的查找函数的写法

来源:互联网 发布:赤壁网络花店 编辑:程序博客网 时间:2024/05/24 06:09
 相信大家都会写并查集的查找父亲的函数,然而今天下午,因为这个函数T了将近两个小时,期间换了各种姿势,都是T,最后才想到是查找函数出了问题,为什么呢,先来看看我开始写的函数:
int find(int x)
{
    if(p[x]==x)
        return x;
    return  find(p[x]);
}
我敢保证这是一个毫无问题的函数,但是T了,为什么呢,因为在反复递归的过程中消耗了太多时间,说简单点,就是找了很多次父亲,这让我想起了挑战程序设计上写的并查集的退化问题,因为一旦处理不好退化,这一系列的父子关系就将变成一个线性的,而每次查找都将耗费几乎遍历完所有点的时间,针对这个问题,挑战程序设计上讲了种加rank数组的方法, 但是这里有种更好的方法就是路径压缩,就是当查找的这个儿子的父亲不是他本人的时候,就继续查找,最后把查找的父亲直接赋予给这个儿子,这样以后再查找的相同儿子的情况时,就不用再来依次遍历了,只需一次递归就找到了,下面是这个函数的正确写法,以后可以都这么写,但对于有些问题,还是不能盲目路径压缩的。
正确写法:
 
int find(int x)
{
    if(p[x]==x)
        return x;
    return  p[x] = find(p[x]);(这里是关键!!!!!!)
}
0 0
原创粉丝点击