牛客补题找朋友

hesorchen

于 2020-08-14 12:10:35 发布

阅读量3.9k

点赞数

CC 4.0 BY-SA版权

分类专栏：题解

本文链接：https://blog.youkuaiyun.com/hesorchen/article/details/108001931

题解专栏收录该内容

178 篇文章

订阅专栏

本文深入探讨了并查集算法的实现，通过具体代码展示了如何初始化并查集、查找元素所属集合以及合并两个集合的过程。适用于解决寻找连通组件、判断元素是否属于同一集合等问题。

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

找朋友

并查集

ll fa[50010];

void init()
{
    for (int i = 1; i <= 50000; i++)
        fa[i] = i;
}

ll find(ll a)
{
    if (fa[a] == a)
        return a;
    return fa[a] = find(fa[a]);
}

void mix(ll a, ll b)
{
    ll fa1 = find(a), fb = find(b);
    fa[fb] = fa1;
}

int main()
{
    init();
    ll n, m, p;
    cin >> n >> m >> p;
    for (int i = 1; i <= m; i++)
    {
        ll temp1, temp2;
        cin >> temp1 >> temp2;
        mix(temp1, temp2);
    }
    while (p--)
    {
        ll a, b;
        cin >> a >> b;
        if (find(a) == find(b))
            cout << "Yes\n";
        else
            cout << "No\n";
    }

    return 0;
}