题解 | #连通图#

连通图

https://www.nowcoder.com/practice/569e89823a5141fe8a11ab7d4da21edf

#include <iostream>

using namespace std;
#define N 1000

int father[N];//每个元素父亲下标
int height[N];

void Init(int n) {
    //最开始每一个元素单独构建一个集合
    for (int i = 1; i <= n; i++) {
        father[i] = i;//每个元素都是自己的根
        height[i] = 1;
    }
}

int find(int x) {
    if (x != father[x]) {
        //他还有父亲
        //优化,压缩路径,找到祖先后先不返回,而是设置为自己的新父亲
        father[x] = find(father[x]);
    }
    //x是树的根
    return father[x];
}

void Union(int x, int y, int &num) {
    //合并两棵树,也可以路径压缩,大树吞并小树
    int X = find(x);
    int Y = find(y);
    if (X != Y) {
        num--;
    }
    if (height[X] < height[Y]) {
        father[X] = Y;
    } else if (height[X] > height[Y]) {
        father[Y] = X;
    } else {
        father[Y] = X;
        height[X]++;
    }

}

int main() {
    int n, m;
    while (scanf("%d %d", &n, &m) != EOF) {
        if (n == 0 && m == 0) {
            break;
        }
        Init(n);
        int num = n;
        for (int i = 0; i < m; i++) {
            int x, y;
            scanf("%d%d", &x, &y);
            Union(x, y, num);
        }
        if(num==1){
            printf("YES\n");
        }else{
            printf("NO\n");
        }

    }
    return 0;
}

全部评论

相关推荐

jack_miller:杜:你不用我那你就用我的美赞臣
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务