图论算法----并查集中的路径压缩

一、算法知识

并查集是一种树型的高级数据结构,用于处理集合的合并和查询的问题,应用十分广泛。

因为主要用合并和查询,所以叫做并查集。但是要注意,这里的集合是不能相交的。

并查集主要有两个函数:find(a)和union(a,b)。

find(a)主要用于查询a所在树的根节点。

union(a,b)主要用于合并两棵树。


来看一道例题:

二、题目描述及分析

亲戚

若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易。 现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。 我们规定:如果x和y是亲戚,y和z是亲戚,那么x和z也是亲戚;如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。

输入

第一行:三个整数n,m,p,(n≤5000,m≤5000,p≤5000),分别表示有n个人,m个亲戚关系,询问p对亲戚关系。 以下m行:每行两个数Mi,Mj,1≤Mi,Mj≤N,表示Ai和Bi具有亲戚关系。 接下来p行:每行两个数Pi,Pj,询问Pi和Pj是否具有亲戚关系。

输出

P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。

样例输入

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

样例输出

Yes
Yes
No

记得我以前写过一个关于并查集的博客(并查集算法—-犯罪团伙(黑科技))。里面主要写了并查集的合并的过程。但算法是伪并查集算法,所以很耗时,实际上真正的并查集算法快得多。


以前的代码:

#include<cstdio>
int b[5005];
int main()
{
    int m,n,i,j,x,y,k,p;
    scanf("%d%d%d",&n,&m,&p);
    for(i=1;i<=n;i++)
        b[i]=i;
    for(i=1;i<=m;i++){
        scanf("%d%d",&x,&y);
        k=b[y];
        for(j=1;j<=n;j++){
            if(b[j]==k){
                b[j]=b[x];
            }
        }
    }
    for(i=1;i<=p;i++){
        scanf("%d%d",&x,&y);
        if(b[x]==b[y])
            printf("Yes\n");
        else
            printf("No\n");
    }
}

总体思路:
1、我们把每一个节点所在的树的根节点存入f[]数组中。
2、输入两个数,查找他们的根节点。
3、对他们所在的树的根节点进行合并。
4、最后输入要查找的两个节点,找到他们的根节点,判断他们的根节点是否相同。


但是这种算法的速度也很慢,因为输入数据有可能形成一条单链
,把所有的时间都浪费在find()函数上。


所以我们对find()进行了优化。
优化前:find(a)只把a这一个节点接在了根节点上。

int find(x)
{
     while(parent[x]!=-1)
         x=parent[x];
      return x;
}

优化后:find(a)把a点到根节点的之间的所有节点(包括a)都接到了根节点的下方。这就是路径压缩。
优化后find()函数的主要内容:

int find(int x)
{
    if(!f[x])
        return x;
    return f[x]=find(f[x]);
}



接下来就是union()函数了。其实根本不用写函数,就一条语句:

f[x]=y;



全部代码:

#include<cstdio>
int n,m,c;
int f[5005];
int find(int x)
{
    if(!f[x])
        return x;
    return f[x]=find(f[x]);
}
int main()
{
    int a,b,p,q,i;
    scanf("%d%d%d",&n,&m,&c);
    for(i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        p=find(a);q=find(b);
        if(p!=q)
            f[p]=q;
    }
    for(i=1;i<=c;i++){
        scanf("%d%d",&a,&b);
        if(find(a)==find(b))
            printf("Yes\n");
        else
            printf("No\n");
    }
}


其实还有一种启发式合并的方法,也可以让find()函数的时间缩短。

    原文作者:犯罪团伙问题
    原文地址: https://blog.csdn.net/C20180602_csq/article/details/56280694
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞