并查集 (DSU)

2

学习链接

注:

在合并完所有的节点之后
加一句 for(int i = 0; i < n; i++) Find(i);
将所有的节点的根节点查找一遍,这样能保证所有的同根节点的指向都是根节点 ,不然会导致一些节点所存的节点还位于某个分支节点,这时并不指向根节点,在求同根节点的节点数时会导致错误。

路径压缩参考博客
非递归路径压缩

int Find(int x)
{
	int r = x, j, k = x;
	while(r != bcj[r]) r = bcj[r]; // 寻找根节点 
	while(k != r)  
	{
		j = bcj[k];  // 暂存此时k的父节点 
		bcj[k] = r;  //  将k的父节点改为根节点 
		k = j;  //  k移到父节点   直至全改为根节点  
	}
	return r;
}

递归路径压缩

int Findd(int x)
{
	return bcj[x] == x ? x : bcj[x] = Findd(bcj[x]); 
}

合并

void Union(int x,int y)
{
	x = Findd(x);//  x变成x的根节点 
	y = Findd(y);//   y变成y的根节点 
	if(x != y) bcj[x] = y; //  根节点不同  x的根节点改成y 
}

注:记得更新根节点
在合并完所有节点之后
加一句 for(int i = 0; i < n; i++) Find(i);
这样能保证所有的节点都指向根节点,不然还会一些节点指向的分支节点,在求位于同根节点的节点数时会出错。
(虽然在求这个节点的根节点时,通过Find函数的路径压缩,会求出根节点,但是求之前保存的节点并不是根节点,所有要求一下)
eg:poj 1611
ac代码

#include<iostream>
#include<cstdio>
using namespace std;
const int Max = (int)1e5 + 9; 
int t,n,m;
int bcj[Max];
int Find(int x)
{
	int r = x, j, k = x;
	while(r != bcj[r]) r = bcj[r]; // 寻找根节点 
	while(k != r)  
	{
		j = bcj[k];  // 暂存此时k的父节点 
		bcj[k] = r;  //  将k的父节点改为根节点 
		k = j;  //  k移到父节点   直至全改为根节点  
	}
	return r;
}
int Findd(int x)
{
	return bcj[x] == x ? x : bcj[x] = Findd(bcj[x]); 
}
void Union(int x,int y)
{
	x = Findd(x);//  x变成x的根节点 
	y = Findd(y);//   y变成y的根节点 
	
	if(x != y) bcj[x] = y;
}
/*
if(x == 0 && y != 0) bcj[y] = x;
	else if(y == 0 && x != 0) bcj[x] = y;
*/
void work()
{	
	for(int i = 0; i < n; i++) bcj[i] = i;
	int x, y, sum = 0, t;
	while(m--)
	{
		scanf("%d %d", &t, &x);
		for(int i = 1; i < t; i++)
		{
			scanf(" %d", &y); Union(x, y);
		}
	}
	
	for(int i = 0; i < n; i++) Find(i);// 更新根节点
	int p = bcj[0];//  保存一下0号嫌疑人的根节点
	
	for(int i = 0; i < n; i++)
	{
		if(bcj[i] == p) sum++;// 相同就是嫌疑人
		//printf("%d %d\n",i,bcj[i]);
	}
	cout << sum << endl;
}
int main()
{
	
	while(scanf("%d %d", &n, &m), m + n) 
		work();
	return 0;
}

/
/
/
/
/
/

种类并查集

解决本题有两种方案,一种是带权并查集,一种是完全合并(开二倍或三倍数组)
/
/
/
/
/
/

学习博客
学习链接
例题
sdnu 1076
题目:已知有 n 个学生,其中有男生也有女生,给定 m 个关系,每个关系表示两个同学性别是否相同。如 同学A 和同学 B 是同性,或不是同性,判断这些条件中是否存在矛盾。
Input
第一行为两个整数 n(1 <= n <= 1000), m(1 <= m <= 1000),每个同学编号为1到n,之后m行,每行三个整数,前两个整数表示两个同学的编号,第三个整数为0或者1,若为0,则表示两个同学为同性,否则为异性。
Output
一行,若存在矛盾,输出YES,否则输出NO

5 5
1 2 1
2 3 1
3 4 1
4 5 1
5 1 0

NO

#include<bits/stdc++.h>
using namespace std;
int n, m, bcj[2009];
int x, y ,z;
bool f=0;
int find(int x)
{
	return bcj[x] == x ? x : bcj[x] = find(bcj[x]);
}
void Union(int x, int y)
{
	x = find(x);
	y = find(y);
	if(x != y) bcj[y]=x;
}
void work()
{
	f=0;
	for(int i = 1; i <= n*2; ++i) bcj[i] = i;
	while(m--)
	{
		cin >> x >> y >> z;
		if(x<=0||x>n||y<=0||y>n) f=1;
		if(z==1)
		{
			if(find(x)==find(y)||find(x+n)==find(y+n)) f=1;
			// [1,n], [n+1,2n]
			//  两个并查集建立关系就是敌对
			Union(x,y+n);//   x 是 y 的敌人
			Union(x+n,y);//   y 是 x 的敌人
		}
		else 
		{
			if(find(x)==find(y+n)||find(x+n)==find(y)) f=1;	
			//  同一个并查集建立关系就是朋友
			Union(x,y);
			Union(x+n,y+n);
		}
	}
	if(f) cout << "YES\n";
	else cout << "NO\n";
}
int main()
{
	cin >> n >> m;
	 work();
	return 0;
}

P1525 [NOIP2010 提高组] 关押罪犯

两倍的数组相当于两个监狱,A和B
每次进行这个合并
merge(x, y+n); // 敌人的敌人是朋友
merge(x+n, y);
并没有一定把某个人放到确定监狱,它是用来维护对立关系
数据:
4 6
1 4 2534
2 3 3512
1 2 28351
1 3 6618
2 4 1805
3 4 12884

排完序:
1 2 28351
3 4 12884
1 3 6618
2 3 3512
1 4 2534
2 4 1805

先维护1,2是对立,然后是3,4,但是他们具体在哪个监狱是不确定的,只是保证对立关系,维护敌人的敌人是朋友
然后加入1,3,因为一直在维护敌人的敌人是朋友这个关系,1和2,3是敌人,所以2,3合并在一起了;3和1,4是敌人,所以1,4合并在一起了

然后是2,3,无法调和输出最小矛盾

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
using namespace std;
const int MAXN = 2e5 + 7;
struct node
{
	int x, y, c;
}a[MAXN];
int pre[MAXN];
int n, m;
void init()
{
	for(int i = 1; i <= 2 * n; i++)  // 记得并查集初始化两倍大小
		pre[i] = i;
}

bool cmp(node a, node b)
{
	return a.c > b.c;
}


int find(int x)
{
	if(pre[x] == x)
		return pre[x];
	else
		return pre[x] = find(pre[x]);
}

void merge(int a, int b)
{
	int x = find(a);
	int y = find(b);
	if(x != y)
	{
		pre[y] = x;
	}
	return ;
}

int main()
{
	cin >> n >> m;
	init();
	for(int i = 1; i <= m; i++)
		scanf("%d%d%d",&a[i].x, &a[i].y, &a[i].c);
	sort(a+1, a+1+m, cmp);
	int flag = 0;
	for(int i = 1; i <= m; i++)
	{
		int x = a[i].x;
		int y = a[i].y;
		int c = a[i].c;
		if(find(x) == find(y)) // 两个罪犯在一个并查集之中 不可调和
		{
			cout << c << endl;
			return 0;
		}
		merge(x, y+n); // 敌人的敌人是朋友
		merge(x+n, y);
	}
	cout << 0 << endl;  // 特判
	return 0;
}


食物链

用一个三倍大小的并查集进行维护,用i+n表示i的捕食对象,而i+2n表示i的天敌。

#include<bits/stdc++.h>
using namespace std;
const int N = 50009;
int bcj[N*3];
// [1,n] [n+1-2n] [2n+1,3n]
// x+n吃x,x+2n吃x+n,x吃x+2n

int n, k;
int d, x, y;
int ans;
int Find(int x)//  递归写法在sdnuoj会超时
{
	int r = x, j, k = x;
	while(bcj[r] != r) r = bcj[r];
	while(k != r)
	{
		j = bcj[k];
		bcj[k] = r;
		k = j;
	}
	return r;
}
void Union(int x, int y)
{
	x = Find(x);
	y = Find(y);
	if(x != y) bcj[y] = x;
}
int main()
{
	cin >> n >> k;
	for(int i = 1;i <= n*3; ++i) bcj[i] = i;
	
	while(k--)
	{
		scanf("%d %d %d", &d,&x,&y);
		if (x > n || y > n || x < 1 || y < 1) { ans++;continue; }
		if(d == 1)
		{
			if(Find(x) == Find(y+n) || Find(x) == Find(y+2*n)) 
			{
				ans++;continue;
			}
			Union(x, y);
			Union(x+n,y+n);
			Union(x+2*n,y+2*n);
		}
		else 
		{
			if(Find(x) == Find(y) || Find(x) == Find(y+2*n))
			{
				ans++;continue;
			}
			Union(x,y+n);
			Union(x+n,y+2*n);
			Union(x+2*n,y);
		}
	}
	cout << ans << endl;
	return 0;
}

sdnu oj 1016
矩形合并

#include<bits/stdc++.h>
using namespace std;
int bcj[109];
struct jx
{
	int x1,x2,y1,y2;
}a[109];
bool check(jx a, jx b)// 这个check是灵魂   画图模拟一下
{
// a的左下角的坐标大于b的右上角  保证开始有重叠
// 然后限定  a的右上角大于b的左下角
	if(a.x1 < b.x2 && a.y1 < b.y2 && a.x2 > b.x1 && a.y2 > b.y1)
		return 1;
	return 0;
}
int Find(int x)
{
	return x == bcj[x] ? x : bcj[x] = Find(bcj[x]);
}
Union(int x, int y)
{
	bcj[Find(x)] = bcj[Find(y)];
}
int main()
{
	int n;
	int sum = 0;
	cin >> n;
	for(int i = 1; i <= n; ++i)
		bcj[i] = i;
	for(int i = 1; i <= n; ++i)
		scanf("%d %d %d %d", &a[i].x1,&a[i].y1,&a[i].x2,&a[i].y2);
	for(int i = 1; i <= n; ++i)
	{
		for(int j = i + 1; j <= n; ++j)
		{
			if(check(a[i],a[j]))
				Union(i, j);
		}
	}
	int ans = 0;
	for(int i = 1; i <= n; ++i)
		if(bcj[i] == i) ans++;
	cout << ans << endl;
	return 0;
}

牛客水题~
至于贴一下这个题的原因嘛。。。
就是老师这个点是迷惑性的

#include<bits/stdc++.h>
using namespace std;
int m, n;
int bcj[60];
int Find(int x)
{
	return x == bcj[x] ? x : bcj[x] = Find(bcj[x]);
}
void Union(int x,int y)
{
	x = Find(x);
	y = Find(y);
	if(x != y)
		bcj[y] = x;
}
int main()
{
	while(~scanf("%d", &n))
	{
		for(int i = 1; i <= 55; ++i) bcj[i] = i;
		cin >> m;
		int x, y;
		for(int j = 1; j <= m; ++j)
		{
			cin >> x >> y;
			if(x != n) Union(x, y);// 老师有学生的手机号没用,不合并
		}
		/*for(int i = 1; i <= n; ++i)
			Find(i);
		for(int i = 1; i <= n; ++i)
			cout << bcj[i] << " "; cout << endl;*/
		if(Find(1) == Find(n))
			cout << "Yes\n";
		else cout << "No\n";
	}
	return 0;
}

/*
4 3
2 3
1 4
4 1

4 3
2 3
4 1
1 4
*/
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值