PAT_甲级_1118 Birds in Forest

题目大意

现在有N张图片,每一张图片里面有K只鸟,在同一张图片中的鸟属于同一棵树,计算森林中树木的最大数目,并且对于任意一对鸟,判断是否在同一棵树上。

算法思路

本题考查并查集的应用,我们使用set集合birds保存所有的输入的鸟,并在输入每一张图片的时候,将其中所有的鸟进行合并为一组,然后对于birds中所有的鸟类根据其祖先归并为一棵树,并存放到set集合trees中,最后对于任意两个鸟的编号,只需要比较father数组的值是否相同即可。

注意点

  • 1、查找father的方法得使用路径压缩,不然会有一个测试点超时。

提交结果

PAT_甲级_1118 Birds in Forest_第1张图片

AC代码

#include
#include
#include
#include

using namespace std;

unordered_set birds;
unordered_set trees;

int father[10005];

int findFather(int x){
    int a = x;
    while(x!=father[x]){
        x = father[x];
    }
    while (a!=father[a]){
        int z = father[a];
        father[a] = x;
        a = z;
    }
    return x;
}

void Union(int a,int b){
    int fa = findFather(a);
    int fb = findFather(b);
    if(fa!=fb){
        father[fa] = fb;
    }
}

int main() {
    for(int i=0;i<=10000;++i){
        father[i] = i;
    }
    int n;
    scanf("%d",&n);
    for(int i=0;i t(k);
        for(int j=0;j

你可能感兴趣的:(c++数据结构和算法)