并查集的迭代优化-食物链

#include<cstdio>
#include<iostream>
using namespace std;
int set[300001];


int findx(int a) {  
    
    if (set[a]==a) return a;  
    int t=a;  
    while (set[a]!=a) {  
        a=set[a];  
    }  
    while (set[t]!=t) {  
        int t2=t;  
        t=set[t];  
        set[t2]=a;  
    }  
    return a;  


    }  
bool same(int x,int y){
return findx(x)==findx(y);
}
int unionn(int x,int y){
int fx=findx(x);
int fy=findx(y);
set[fx]=fy;
}
int main(){
int n;
int k;
cin>>n>>k;

for(int i=1;i<=3*n;i++){
set[i]=i;
}
int a,b,c;
int cnt=0;

for(int i=1;i<=k;i++){
cin>>a>>b>>c;

if(b>n||c>n){
cnt++;
continue;
}
if(a==1){
if(same(b,c+n)||same(b,c+2*n)){
cnt++;
continue;
}
else{
unionn(b,c);
unionn(b+n,c+n);
unionn(b+2*n,c+2*n);
}
}
else if(same(b,c)||same(b,c+2*n)){
cnt++;
continue;
}
else{
unionn(b,c+n);
unionn(b+n,c+2*n);
unionn(b+2*n,c);
}



}
cout<<cnt<<endl;
}