05-树8 File Transfer

　　课上例题。

　　这题考查对并查集的运用以及对之前并查算法的优化。

#include <stdio.h>

#define MAXN 10000                    /* 集合最大元素个数 */

typedef int ElementType;            
typedef ElementType SetName;        /* 默认用根节点的下标作为结点名称 */
typedef ElementType SetType[MAXN];    /* 假设集合元素下标从 0 开始 */

void Initialization(SetType S, int N);
SetName Find(SetType S, ElementType X);
void Union(SetType S, SetName Root1, SetName Root2);
void Input_connection(SetType S);
void Check_connection(SetType S);
void Check_netWork(SetType S, int N);

int main()
{
    SetType S;
    int N;
    char in;
    scanf("%d
", &N);
    Initialization( S, N );
    
    do {
        scanf("%c", &in);
        switch ( in ) {
            case 'I':    Input_connection( S );    break;
            case 'C':    Check_connection( S );    break;
            case 'S':    Check_netWork( S, N );    break;
        }
    } while ( in != 'S' );
    
    return 0;
}

void Input_connection(SetType S)
{
    ElementType x1, x2;
    SetName Root1, Root2;
    
    scanf("%d %d
", &x1, &x2);
    Root1 = Find(S, x1-1);
    Root2 = Find(S, x2-1);
    
     if ( Root1 != Root2 )
         Union(S, Root1, Root2);
}

void Check_connection(SetType S)
{    /* 检查 是否 连通 */
    ElementType x1, x2;
    SetName Root1, Root2;
    
    scanf("%d %d
", &x1, &x2);
    Root1 = Find(S, x1-1);
    Root2 = Find(S, x2-1);
    
    if ( Root1 == Root2 ) printf("yes
");
    else                   printf("no
");
    
}

void Check_netWork(SetType S, int N)
{    /* 检查 连通性 */
    int i, cnt = 0;
    
    for ( i = 0; i < N; i++ ) {
        if ( S[i] < 0 )        cnt++; 
    }    
    
    if ( cnt == 1 )
        printf("The network is connected.
");
    else 
        printf("There are %d components.
", cnt);
}

void Union(SetType S, SetName Root1, SetName Root2)
{
    /* 保证 小集合 并入 大集合 */
    if ( S[Root1] < S[Root2] ) { /* 如果 Root1 规模大 */
        S[Root1] += S[Root2];
        S[Root2] = Root1;         /* Root2 并入 Root1 */ 
    }
    else {                         /* 如果 Root2 规模大 */
        S[Root2] += S[Root1];
        S[Root1] = Root2;         /* Root1 并入 Root2 */ 
    }
}

SetName Find(SetType S, ElementType X)
{
    if ( S[X] < 0 )  /* 找到集合的根 */ 
        return X;
    else
        return S[X] = Find(S, S[X]); /* 路径压缩 */ 
}

void Initialization(SetType S, int N)
{    /* 集合元素全部初始化为 -1 */ 
    int i;
    for ( i = 0; i < N; i++ ) 
        S[i] = -1;
}

　　不得不说，这里的 switch 语句用的太漂亮了，赞。