G-04 朋友圈的数量 (20 分) （并查集）

学校有N个学生，许多同学在学习中与生活中成为了朋友，据说朋友关系有自反性（自己和自己是朋友），有对称性（你和我是朋友，我和你一定也是朋友），还有传递性，即如果A和B是朋友，且B和C是朋友，则A和C也是朋友，于是就形成了许多的朋友圈。下列矩阵表示的四个人的朋友圈：i行 j列为1表示是直接的朋友关系，为0表示并非直接的朋友关系，是否是间接的朋友不清楚。输入样例中，0,1,2是一个朋友圈，3号自己是一个朋友圈。故共有2个朋友圈。请编写程序，根据给定的关系矩阵，计算学校的朋友圈数量。 注意：因为关系矩阵是对称的，故只录入了下三角的数据，也就是只描述了i号学生与其前边的学生的朋友关系，后边的关系未再复述（包括对角线上的自己：N个学生，仅输入了N*(N+1)/2个数据）。

输入样例：

4
1
1 1
1 0 1
0 0 0 1

输出样例：

输出全校学生中的朋友圈的数量（指没有交集的朋友圈）：

2

代码如下： 

#include<bits/stdc++.h>
using namespace std;
int p[10005];
int n , ans;

int f(int x)
{
    if(p[x]==x)
        return x;
    else
        return p[x]=f(p[x]);
}

void unite(int x , int y)
{
    x=f(x);
    y=f(y);
    if(x==y) return ;
    p[y]=x;
    //p[x]=y;
}

int main ()
{
    for(int i=0;i<10005;i++)
        p[i]=i;
    cin>>n;
    int k;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<=i;j++)
        {
            cin>>k;
            if(k)
                unite(i,j);
        }
    }
    for(int i=0;i<n;i++)
        if(p[i]==i)
            ans++;
    cout<<ans;
    return 0;
}