題目:https://ac.nowcoder.com/acm/contest/22904/1024
思路:這道題網路上有很多思路,可以開三個並查集,可以使用帶權並查集,但是有一個大佬的思路是這樣的,將總結點的數量增加到3n個,把整個節點區域分為n,2*n,3*n三個部分,我們可以物種a的一個節點對應物種b的兩個節點,如果是同類,我們就把他們透過範圍對應起來(如a_n,a_2*n,a_3*n和b_n,b_2*n,b _3*n對應),如果是不是同類,我們讓這個節點指向下一個範圍的節點(如a_n,a_2*n,a_3*n和b_2*n,b_3*n,b _n對應),如此一來我們只需要判斷不同區域範圍的節點之間的關係即可
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn = 55000;
int a[3*maxn];
void init(int x){
for(int i=0;i<3*x;i++){
a[i] = i;
}
}
int find(int x){
if(a[x] == x) return x;
return a[x] = find(a[x]);
}
void join(int x,int y){
if(find(x) != find(y)) {
a[find(x)] = find(y);
}
}
int main(){
int n,k;
scanf("%d %d",&n,&k);
init(n);
int ans =0;
while(k--){
int d,x,y;
scanf("%d %d %d",&d,&x,&y);
if((x<1 || x>n || y<1 || y > n) || (d==2 && x==y)) {
ans++;
continue;
}
if(d==1) {
if(find(x)==find(y+n) || find(x+n)==find(y)) ans++;
else {
join(x,y);
join(x+n,y+n);
join(x+2*n,y+2*n);
}
}
else if(d==2){
if(find(x)==find(y) || find(x+n)==find(y)) ans++;
else {
join(x,y+n);
join(x+n,y+2*n);
join(x+2*n,y);
}
}
}
printf("%d",ans);
return 0;
}