并查集模板

最新推荐文章于 2021-12-15 00:30:14 发布

冷眼旁观你的泪

最新推荐文章于 2021-12-15 00:30:14 发布

阅读量197

点赞数

CC 4.0 BY-SA版权

分类专栏：并查集 ACM

本文链接：https://blog.youkuaiyun.com/FZdeAmbition/article/details/80033467

ACM 同时被 2 个专栏收录

28 篇文章

订阅专栏

并查集

1 篇文章

订阅专栏

本文详细介绍了并查集算法的基本概念、实现原理及其在解决实际问题中的应用。通过多个实例展示了如何利用并查集进行数据分组、判断连通性和处理动态关系变化等问题。

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

int fa[N],rank[N];
void init(int n) { //并查集初始化
for (int i = 0; i <= n; i++)
fa[i] = i;

rank[i] = 0;//高度设置为0;如果涉及树的高度，P87
}
}
void unite(int u, int v) {//并查集合并操作
int fau = find(u);
int fav = find(v);
if (fau == fav) return;

fa[fav] = fau;

}
int find(int u) {//查找集合编号操作
if (fa[u] != u) {
fa[u] = find(fa[u]);
}
return fa[u];
}
/* 实际上，可以写的更短


int fa[N];
void init(int n) {
for (int i = 0; i <= n; fa[i] = i) fa[i] = i;
}
void unin(int u, int v) {
fa[find(v)] = find(u);
}
int find(int u) {
return fa[u] == u ? fa[u] : fa[u] = find(fa[u]);

对立并查集：POJ1703

#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
const int maxn = 110;
const int INF = 0x3f3f3f3f;
int pre[200005],enemy[200005];
int find(int x)
{
	if(x == pre[x])
		return x;
	return pre[x] = find(pre[x]);
}
void unite(int x,int y)
{
	pre[find(y)] = find(x);
}
int main()
{
	int t;
	cin >> t;
	while(t--)
	{
	    memset(enemy,0,sizeof(enemy));
		int m,n;
		cin >> n >> m;
		for(int j = 1;j <= n;j++)
		{
			pre[j] = j;
		}
		cin >> n >> m;
		for(int i = 0;i < m;i++)
		{
			getchar();
			char temp;
			int a,b;
			scanf("%c%d%d",&temp,&a,&b);
			if(temp == 'D')
			{
				if(enemy[a])//如果a有敌人，那么a的敌人肯定和b是一伙
					unite(enemy[a],b);
				if(enemy[b])//如果b有敌人，那么b的敌人肯定和a是一伙
					unite(enemy[b],a);
				enemy[a] = b;//转换敌人
				enemy[b] = a;
			}
			else
			{
				if(find(a) == find(enemy[b]))
					cout << "In different gangs." << endl;
				else if(find(a) == find(b))
					cout << "In the same gang." << endl;
				else
					cout << "Not sure yet." << endl;
			}
		}
	}
}

用一个数组的做法：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1e5 + 5;
int n, m;
int par[maxn*2];
void init(int n)
{
    for(int i = 1; i <= n; i++)
    {
        par[i] = i;
    }
}

int getRoot(int x)
{
    return par[x] == x ? x : par[x] = getRoot(par[x]);
}

void unite(int x, int y)
{
    x = getRoot(x);
    y = getRoot(y);
    if(x == y)  return;
    par[y] = x;
}

bool same(int x, int y)
{
    return getRoot(x) == getRoot(y);
}

int main()
{
    //freopen("in.txt", "r", stdin);
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &n, &m);
        init(2*n);

        char op;    int a, b;
        for(int i = 1; i <= m; i++)
        {
            scanf("\n%c%d%d", &op, &a, &b);
            if(op == 'D')
            {
                unite(a, b+n);
                unite(a+n, b);
            }
            else
            {
                if(same(a, b+n) || same(a+n, b))    printf("In different gangs.\n");
                else if(same(a, b) || same(a+n, b+n))   printf("In the same gang.\n");
                else    printf("Not sure yet.\n");
            }
        }
    }
    return 0;
}

poj2236修电脑问题

思路是把能够通信的放在一组里，然后最后看是否在一个父结点。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1e5 + 5;
int n, m;
int par[maxn*2];
int visit[maxn*2];
int x[maxn*2];
int y[maxn*2];
int d;
void init(int n)
{
    for(int i = 1; i <= n; i++)
    {
        par[i] = i;
        visit[i] = 0;
    }
}
int find(int x)
{
    return par[x] == x ? x : par[x] = find(par[x]);
}

void unite(int x, int y)
{
    x = find(x);
    y = find(y);
    if(x == y)  return;
    par[y] = x;
}
bool check(int a,int b)
{
    if((x[a]-x[b])*(x[a]-x[b]) + (y[a]-y[b])*(y[a]-y[b]) <= d*d)
    {
        return true;
    }
    return false;
}
bool same(int x, int y)
{
    return find(x) == find(y);
}

int main()
{
    int n;
    cin >> n >> d;
    for(int i = 1;i <= n;i++)
    {
        cin >> x[i] >> y[i];
    }
    init(n);
    char c;
    int temp;
    while(cin >> c)
    {
        if(c == 'O')
        {
            int r;
            scanf("%d",&r);
            visit[r] = 1;
            for(int i = 1; i <= n; i++)
                if(i!=r && visit[i] && check(r,i))
            {
                unite(r,i);
            }
        }
        if(c == 'S')
        {
            int r,l;
            scanf("%d%d",&r,&l);
            int k1 = find(r);
            int k2 = find(l);
            if(k1 == k2)
            {
                cout << "SUCCESS" << endl;
            }
            else
            {
                cout << "FAIL" << endl;
            }
        }
    }
}