并查集有两个扩展应用,一个是带边权的并查集另一个是带扩展域的并查集。这道题用两种方法都可以做。值得学习。
题目:奇偶游戏
做法1:带边权的并查集
1、用
s
u
m
[
x
]
sum[x]
sum[x] 表示
1
1
1到
x
x
x 中1的个数(此处
s
u
m
[
x
]
sum[x]
sum[x] 不需要具体求出),读入“
l
,
r
,
r
e
l
l,r,rel
l,r,rel”,如果
r
e
l
=
o
d
d
rel=odd
rel=odd,那么证明
s
u
m
[
l
−
1
]
sum[l-1]
sum[l−1] 和
s
u
m
[
r
]
sum[r]
sum[r] 奇偶性不同(前缀和的思想),反之则相同。
2、有了这个操作,我们就可以把输入数据当作 x 1 , x 2 , . . . x_1,x_2,... x1,x2,... 和他们奇偶性的关系,如果是奇偶性不同,两个元素之间则连接一条边权为 1 的边反之为 0 。此时,记 d [ x ] d[x] d[x] 为 x x x 到 p [ x ] p[x] p[x] 的奇偶性。同时,这些元素的关系具有传递性,可以相互推出,所以把可以推出的元素放在一个集合里,用并查集维护 d [ x ] d[x] d[x]。
3、接下来并查集维护时面临两个问题:
a)调用 find() 函数时子节点直接指向根节点过程中边权的计算;
本质上就是亦或运算(
x
o
r
xor
xor),因为相同出 0,不同出 1,即奇+奇=偶,偶+偶=偶,奇+偶=奇,这也与上面的定义正好一致。
b)合并两颗树时加的那一条边的边权的计算;(以
p
[
x
]
=
f
i
n
d
(
y
)
p[x]=find(y)
p[x]=find(y) 操作说明)
这里不难看出公式
r
e
l
(
x
,
y
)
=
d
[
x
]
rel(x,y)=d[x]
rel(x,y)=d[x]
x
o
r
xor
xor
d
[
y
]
d[y]
d[y]
x
o
r
xor
xor
d
[
p
[
x
]
]
d[p[x]]
d[p[x]],则可以得到新加的边权计算公式为:
d
[
p
[
x
]
]
=
d
[
x
]
d[p[x]]=d[x]
d[p[x]]=d[x]
x
o
r
xor
xor
d
[
y
]
d[y]
d[y]
x
o
r
xor
xor
r
e
l
(
x
,
y
)
rel(x,y)
rel(x,y).
4、最后要注意离散化数据,这里不要求保序性,所以用偷个懒用 m a p map map 好了。
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <unordered_map>
using namespace std;
const int N=10010;
unordered_map<int,int> mp;
int p[N*2],d[N*2];
struct node{
int x,y,a;
}q[N];
int n,m,tot;
int dis(int x){
if(mp.count(x)==0) mp[x]=++tot;
return mp[x];
}
int find(int x){
if(x!=p[x]){
int root=find(p[x]);
d[x]^=d[p[x]];
p[x]=root;
}
return p[x];
}
int get(char x[]){
if(x[0]=='o') return 1;
else return 0;
}
int main(){
cin>>n>>m;
for(int i=1;i<=m;i++){
int x,y;
char s[5];
scanf("%d%d%s",&x,&y,s);
int b=get(s);
q[i].x=dis(x-1); //注意是x-1
q[i].y=dis(y);
q[i].a=b;
}
for(int i=1;i<=tot;i++) p[i]=i;
for(int i=1;i<=m;i++){
int pa=find(q[i].x),pb=find(q[i].y);
if(pa!=pb){
p[pa]=pb,d[pa]=d[q[i].y]^q[i].a^d[q[i].x];
}
else{
if(d[q[i].x]^d[q[i].y]!=q[i].a){
cout<<i-1<<endl;
return 0;
}
}
}
cout<<m<<endl;
return 0;
}