C124 扩展域并查集 P2024 [NOI2001] 食物链

C124 扩展域并查集 P2024 [NOI2001] 食物链_哔哩哔哩_bilibili

 

P2024 [NOI2001] 食物链 - 洛谷

题目：三类动物 A,B,C，这三类动物的食物链构成了有趣的环形。A 吃 B，B 吃 C，C 吃 A。有 N 个动物，K 句话。求假话的总数。

// 扩展域并查集 
#include<bits/stdc++.h>
using namespace std;

int n,k,ans;
int fa[50005*3];

int find(int x){
  return fa[x]==x?x:fa[x]=find(fa[x]);
}
void unite(int x,int y){
  fa[find(y)]=find(x);
}
int main(){
  scanf("%d%d",&n,&k);
  for(int i=1; i<=3*n; i++) fa[i]=i; //同类域[1,n] 猎物域[n+1,2n] 天敌域[2n+1,3n]
  for(int i=1,op,x,y; i<=k; i++){
    scanf("%d%d%d",&op,&x,&y);
    if(x>n||y>n) ans++;
    else if(op==1){ //x,y是同类
      if(find(x)==find(y+n)||find(x)==find(y+n+n))ans++; //如果x在y的猎物域,或x在y的天敌域,是假话ans+1
      else{ //是真话,合并同类集合
        unite(x,y);
        unite(x+n,y+n);
        unite(x+n+n,y+n+n);
      }
    }
    else{ //x吃y
      if(find(x)==find(y)||find(x)==find(y+n))ans++; //如果他们在同类域,或者x在y的猎物域,是假话ans+1
      else{ //是真话,合并同类集合
        unite(x,y+n+n); 
        unite(x+n,y);   
        unite(x+n+n,y+n); //x吃y,y吃z,z吃x
      }
    }
  }
  printf("%d\n",ans);
}