注:
在合并完所有的节点之后
加一句 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;
}
两倍的数组相当于两个监狱,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
*/