并查集模板

并查集：判断几个人是否是同一家庭的人

 

初始化：for(i=1;i<=n;i++)
               {
                       p[i]=i;     
                       t[i]=0;     //家庭中的人数
                }

 

找祖先（如果有联系就会有同一个祖先）：int find1(int x)

                                                                            {
                                                                                  return p[x]==x?x:find1(p[x]);
                                                                             }

 

如果有联系就入祖籍： for(i=1;i<=m;i++)
                                         {
                                               int a=find1(x[i]);
                                               int b=find1(y[i]);
                                               if(a!=b)
                                               p[a]=b;

                                         }

举栗子：

输入n个人，m种联系    输出人数最多的家族，并按从小到大的顺序输出家族人员

 

#include <iostream>
#include <cstdio>

using namespace std;

int x[205],y[205],z[205],p[205],t[205];
int find1(int x)
{
    return p[x]==x?x:find1(p[x]);
}

int main()
{
    int n,m,k,i;
    cin>>n>>m>>k;
    for(i=1;i<=m;i++)
        cin>>x[i]>>y[i]>>z[i];
    for(i=1;i<=n;i++)
    {
        p[i]=i;
        t[i]=0;
    }
    for(i=1;i<=m;i++)
    {
        int a,b;
        if(z[i]==1)
        {
            a=find1(x[i]);
            b=find1(y[i]);
            if(a!=b)
                p[a]=b;
        }
    }
    while(k--)
    {
        int s1,s2;
        cin>>s1>>s2;
        for(i=1;i<=m;i++)
        {
            if(z[i]==-1&&x[i]==s1&&y[i]==s2||z[i]==-1&&x[i]==s2&&y[i]==s1)
            {
                if(find1(s1)==find1(s2))
                    {cout<<"OK but..."<<endl;
                     break;
                    }
                else
                    {cout<<"No way"<<endl;
                     break;
                    }
            }
            else
            {
                if(find1(s1)==find1(s2)&&i==m)
                    cout<<"No problem"<<endl;
                else if(find1(s1)!=find1(s2)&&i==m)
                    cout<<"OK"<<endl;

            }
        }
    }
    return 0;
}