前置技能:并查集
拓展域并查集解决了一种多个有相互关系的并查集,放在一起考虑的问题。一般的并查集应用一般就是判断在不在一个集合,拓展域并查集讲的是多个集合,之间有相互关系一般为相互排斥关系,判断是否在一个集合等。
首先对与最简单的并查集来说,如果两个是同一类,那么就 join(a,b)对吧,但是对于两个相互排斥类的怎么办呢,这就涉及到拓展与并查集了,首先想法就是建立两个并查集,但是怎么把两个并查集联系起来呢------拓展个体。
这里的拓展个体是什么意思呢,一个个体我们要拆成多个,比方说两个集合存在队立关系,那么对于一个个体a,我们假设存在一个个体 a+n ,a和a+n这两个是处于对立关系的,所以当我们说 a 和 b对立的时候,意思就是在说,a + n 和 b在同一并查集,b+n和a在同一并查集,当我们说,a和b是同类的时候,那么也就是说 a和b属于一个并查集,且a+n和b+n属于一个并查集。
这样就建出了多个并查集,解决了多个集合的相互关系。
例题1:食物链
题目链接:http://poj.org/problem?id=1182
这道算是最经典的拓展域并查集题目了,A吃B,B吃C,C吃A,那么我们就需要三个并查集,也就是原数组扩展3倍。
A和B是同类说明了:
1. A和B在同一并查集 。 2.A+n和B+n在同一并查集 。 3.A+2*n和B+2*n在同一并查集。
A吃B说明了:
1.A+n和B是同类。 2.A+2*n和B+n是同类。 3.A和B+2*n是同类。
(搞不清楚可以画一个转圈圈的图,表明A,B , C)
#include<iostream>
#include<stdio.h>
using namespace std;
int fa[200005];
int find1(int x)
{
if(x==fa[x])return x;
else return fa[x]=find1(fa[x]);
}
void join(int x,int y)
{
if(find1(x)!=find1(y))fa[find1(x)]=find1(y);
}
int main()
{
int n,k;
scanf("%d%d",&n,&k);
int ans=0;
int d,x,y;
for(int i=1;i<=3*n;i++)
{
fa[i]=i;
}
while(k--)
{
scanf("%d%d%d",&d,&x,&y);
if(x>n||y>n)
{
ans++;continue;
}
if(d==1)
{
if(find1(x+n)==find1(y)||find1(y+n)==find1(x))ans++;
else join(x,y),join(x+n,y+n),join(x+2*n,y+2*n);
}
if(d==2)
{
if(find1(x)==find1(y)||find1(y+n)==find1(x))ans++;
else join(x+n,y),join(x,y+2*n),join(y+n,x+2*n);
}
}
printf("%d\n",ans);
}
例题2:http://codeforces.com/problemset/problem/766/D
比上一个更简单,只有两个集合,A和B同义,A和B反义(画圈圈)不建议用map字符串,能hash还是hash的好
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int MAXN=1e5+5;
int fa[3*MAXN];
int Find(int x)
{
if(x==fa[x])return x;
return fa[x]=Find(fa[x]);
}
void add(int x,int y){
if(Find(x)!=Find(y))
fa[Find(x)]=Find(y);
}
int main()
{
map<string,int>ss;
int n,m,q,cnt=0;
scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<=2*n;i++){
fa[i]=i;
}
for(int i=1;i<=n;i++){
string s;
cin>>s;
if(!ss[s])ss[s]=++cnt;
}
for(int i=1;i<=m;i++){
int op;
string s1,s2;
scanf("%d",&op);
cin>>s1>>s2;
int a=ss[s1],b=ss[s2];
if(op==1){
if(Find(a+n)==Find(b)||Find(b+n)==Find(a)){
printf("NO\n");
}
else add(a,b),add(a+n,b+n),printf("YES\n");
}
else{
if(Find(a)==Find(b)||Find(a+n)==Find(b+n)){
printf("NO\n");
}
else add(a+n,b),add(a,b+n),printf("YES\n");
}
}
for(int i=1;i<=q;i++){
string s1,s2;
cin>>s1>>s2;
int a=ss[s1],b=ss[s2];
if(Find(a)==Find(b)){
printf("1\n");
}
else if(Find(a+n)==Find(b)){
printf("2\n");
}
else printf("3\n");
}
}