带权并查集--P2024 [NOI2001]食物链

因为我们一共有三个群体，所以我们可以开三倍的并查集，$1$~$n$的部分表示为A的群体，$n+1$~$2n$的部分表示B的群体，$2n+1$~$3n$的部分表示C的群体

题干给的三种关系，A吃B，B吃C，C吃A

我们可以由二倍并查集，由特殊到一般，由二到三，在二倍并查集中：当A中的x与B中的y合并，有关系x吃y，当C中的x与C中的y合并，有关系x和y是同类。那么在三倍并查集中就有这样的几种关系：

udpate(x,y) | update(x+n,y+n) | update(x+n+n,y+n+n)表示着同类合并

find(x)==find(y+n)表示x吃y同理可得剩下的情况（主要是因为我懒）

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
int n,k;
const int maxn=1e5+10;
int mark,x,y;
int ans,fa[3*maxn];
void init(){
  for (int i = 1;i <= n*3;i++) fa[i]=i;
}
int find(int x){
  if (fa[x]==x) return x;
  else return fa[x]=find(fa[x]);
}
void update(int x,int y){
  int fx=find(x),fy=find(y);
  fa[fx]=fy;
}
int main(){
  scanf("%d%d",&n,&k);
  init();
  for (int i = 1;i <= k;i++){
    scanf ("%d%d%d",&mark,&x,&y);
    if (x>n||y>n){ans++; continue;}
    if (mark==1){
      if (find(x)==find(y+n)||find(x+n)==find(y)) ans++;
      else {
    update(x,y);
    update(x+n,y+n);
    update(x+n+n,y+n+n);
      }
    }
    else{
      if (find(x)==find(y)||find(x+n)==find(y)) ans++;
      else {
    update(x,y+n);
    update(x+n,y+n+n);
    update(x+n+n,y);
      }
    }
  }
  printf("%d\n",ans);
  return 0;
}