并查集 按秩合并&路径压缩

并查集自己也只是会了方法,稍微有点思维难度的题就写不出来【例如银河英雄传说】,Menci教的代码。
还有Menci是把并查集写成了一个结构体,里面有各种成员函数,也超级棒。

例题是HDU 1232

#include 

const int maxn = 1005;

int f[maxn], dep[maxn], n, m;

int Find(int x) //带路径压缩 
{
    return x == f[x] ? x : f[x] = Find(f[x]);
}

void Merge(int x, int y) //按秩合并 
{
    int fx = Find(x);
    int fy = Find(y);

    if (dep[fx] <= dep[fy])
    {
        f[fx] = fy;
        if (dep[fx] == dep[fy])
        {
            dep[fy]++;
        }
    }
    else
    {
        f[fy] = fx;
    }
}

int main()
{
    //scanf("%d %d", &n, &m);
    while (scanf("%d %d", &n, &m) != EOF && n)
    {
        for (int i = 1; i <= n; i++)
        {
            f[i] = i;
        }
        for (int i = 1; i <= m; i++)
        {
            int x, y;
            scanf("%d %d", &x, &y);
            Merge(x, y);
        }

        int ans = 0;

        for (int i = 1; i <= n; i++)
        {
            if (f[i] == i)
            {
                ans++;
            }
        }

        printf("%d\n", ans - 1);
    }

    return 0;
} 

这道题是合并之后,只要把树的根都连起来它们就都连通了,我到底是有多弱啊还是没想到Orz

不带按秩合并的merge是这样子

void merge(int x, int y)
{
    a[find(x)] = find(y);
}

你可能感兴趣的:(算法)