检查网络(并查集)

题目要求：

给定一个计算机网络以及机器的双向连线列表，每一条连线允许两端的计算机进行直接的文件传输，其他计算机间若存在一条连通路径，也可以进行间接的文件传输，请写程序判断，任意指定两台计算机，它们之间是否可以进行文件传输？

输入要求：

输入由若干组测试数据组成。对于每一组测试数据，第1行包含一个整数N（<=10000），即网络中计算机的总台数，因而每台计算机可用1到N之间的一个正数表示。接下来的几行输入格式为I C1 C2或者C C1 C2或S，其中C1和C2是两台计算机的序号，I表示在C1和C2间输入一条连线，C表示检查C1和C2间是否可以传输文件，S表示该组测试结束。

当N为0时，表示全部测试结束，不要对该数据做任何处理。

输出要求：

对每一组C开头的数据，检查C1和C2间是否可以传输文件，若可以，则在一行中输出“yes”，否则输出“no”。

当读到S时，检查整个网络。若网络中任意两机器间都可以传输文件，则在一行中输出“The network is connected.”，否则输出“There are k components.”其中k是网络中连通集的个数。

#include <iostream>
using namespace std;
int a[10001],b[10001];
int find(int x)
{
    if(x==a[x])
        return x;
    else
        return(find(a[x]));
}
void unio(int r1,int r2)//合并时进行路径压缩
{
    if(b[r1]>=b[r2]) //r1所在连通集长度大于r2所在连通集
    {
        b[r1]+=b[r2];
        a[r2]=r1;
    }
    else
    {
        b[r2]+=b[r1];
        a[r1]=r2;
    }
}
int main()
{
    int n,u,v,r1,r2;
    char c;
    cout<<"请输入网络中计算机的总台数："<<endl;
    while(cin>>n)
    {
        if(n==0) break;
        for(int i=1; i<=n; i++) //初始化
        {
            a[i]=i;
            b[i]=1;
        }
        cout<<"请输入操作："<<endl;
        while(cin>>c)
        {
            if(c=='I')//为两台电脑之间添加连接
            {
                cin>>u>>v;
                r1=find(u);
                r2=find(v);
                if(r1!=r2) //u和v不连通时
                {
                    unio(r1,r2);
                }
                cout<<"请输入操作："<<endl;
            }
            else if(c=='C')//查询两台机器之间是否连通
            {
                cin>>u>>v;
                r1=find(u);
                r2=find(v);
                if(r1!=r2)
                    cout<<"no"<<endl;
                else
                    cout<<"yes"<<endl;
                cout<<"请输入操作："<<endl;
            }
            else if(c=='S')//检查整个网络并结束该组测试
            {
                int count=0;
                for(int i=1; i<=n; i++)
                    if(a[i]==i)
                        count++;
                if(count==1)//如果整个网络全部连通
                    cout<<"The network is connected."<<endl;
                else
                    cout<<"There are "<<count<<" components."<<endl;
                cout<<endl;
                cout<<"请输入网络中计算机的总台数："<<endl;
                break;
            }
            else
            {
                cout<<"输入错误，请重新输入操作."<<endl;
            }
        }
    }
    return 0;
}