慕课浙大数据结构-05-树8 File Transfer (25分)

忘了参考哪个博客了,并查集,数组下标代表存放的值,数组元素代表其父节点下标,根节点的元素值为负数,其绝对值为含有节点数量

#include<stdio.h>
int s[10005];//forget the ";"
// int find(int x){
    
    
//     for(;s[x]>0;x=s[x]);
//     return x;
// }

int find(int x){
    
    //查找根节点下标
	if(s[x] < 0)
		return x;
	else
		return s[x] = find( s[x] );   //先改变树的结构,再返回根结点
}

// void union(int a,int b){
    
    
//     int root1 = find(a);
//     int root2 = find(b);
//     s[root1] = root2;
// }

void Union(int a, int b) {
    
    //union是关键字 因此这个函数名需要大写 不然会报错
//将a,b两个电脑连在一起,将小集合的树连在大集合的树上
	int faA = find(a);
	int faB = find(b);
	if(s[faA] > s[faB])
		s[faA] = faB;
	else {
    
    
		if(s[faA] == s[faB])
			s[faA]--;
		s[faB] = faA;
	}
}
void Initiate(int n){
    
    
    for(int i=1;i<=n;++i){
    
    
        s[i] = -1;
    }
}
void Input(){
    
    
    int a,b;
    scanf("%d %d",&a,&b);
    Union(a,b);
}
void Check(){
    
    
    int a,b;
    scanf("%d %d",&a,&b);
    int ra = find(a);
    int rb = find(b);
    if(ra == rb){
    
    
        printf("yes\n");
    }
    else{
    
    
    printf("no\n");
    }
}
void Ouput(int n){
    
    
    int cou = 0;
    for(int i=1;i<=n;++i){
    
    
        if (s[i]<0){
    
    
            cou++;
        }
    }
    if(cou == 1){
    
    
        printf("The network is connected.");
    }
    else{
    
    
        printf("There are %d components.",cou);
    }
    
}
int main(){
    
    
    int n;
    char in;
    scanf("%d",&n);
    Initiate(n);
    do{
    
    
        scanf("%c",&in);
        switch(in){
    
    
        case 'I':Input();break;
        case 'C':Check();break;
        case 'S':Ouput(n);break;
        }
    }while(in!='S');
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43919570/article/details/105554105