1118. Birds in Forest (25)

最新推荐文章于 2018-11-13 09:44:29 发布

pacmoc

最新推荐文章于 2018-11-13 09:44:29 发布

阅读量128

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_31474267/article/details/79438400

PAT甲级专栏收录该内容

100 篇文章

订阅专栏

本文深入探讨了并查集算法的实现原理及其在解决特定问题中的应用。通过具体实例介绍了如何利用并查集来判断元素间的关系，并展示了如何通过不断合并集合来减少查找时间，提高算法效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <vector>
#include <algorithm>
#include <stack>
#include <map>
using namespace std;
int father[10001];
int to[10001];

int find(int x){
    while (x != father[x]) {
        x = father[x];
    }
    return x;
}
void Union(int a, int b){
    a = find(a);
    b = find(b);
    if(a < b) father[b] = a;
    else father[a] = b;
}

int main(){
    int n;
    cin >> n;
    map<int, bool> ma;
    vector<int> v;
    for (int i = 0; i < 10001; i++) {
        father[i] = i;
    }
    for (int i = 1; i <= n; i++) {
        int k;
        cin >> k;
        for (int j = 0; j < k; j++) {
            int t;
            cin >> t;
            if(!ma[t]){
                v.push_back(t);
                ma[t] = true;
                to[t] = i;//每一只鸟在那棵树上
            }
            Union(i, find(to[t]));//将两棵树联系在一起
        }
    }
    int cnt = 0;
    map<int, bool> mb;
    for (int i = 0; i < v.size(); i++) {
        if(find(to[v[i]]) && !mb[find(to[v[i]])]){
            mb[find(to[v[i]])] = true;
            cnt++;
        }
    }
    cout << cnt << ' ' << v.size() << endl;

    int k;
    cin >> k;
    for (int i = 0; i < k; i++) {
        int a, b;
        cin >> a >> b;
        if(find(to[a]) == find(to[b])){
            cout << "Yes\n";
        }else{
            cout << "No\n";
        }
    }
    return 0;
}