并查集自己也只是会了方法,稍微有点思维难度的题就写不出来【例如银河英雄传说】,Menci教的代码。
还有Menci是把并查集写成了一个结构体,里面有各种成员函数,也超级棒。
例题是HDU 1232
#include <cstdio>
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);
}