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

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

 

 

P2024 [NOI2001] 食物链 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

// 扩展域并查集 
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

int n,k,op,x,y,ans;
int p[50005*3];
// x是同类域
// x+n是猎物域
// x+n+n是天敌域

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