牛客补题找朋友

原创于 2020-08-14 12:10:35 发布 · 3.9k 阅读

1 ·

CC 4.0 BY-SA版权

题解专栏收录该内容

178 篇文章

订阅专栏

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

找朋友

并查集

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;
}