算法思路:把那些确定了相对关系的节点放在同一棵树里(可以同时存在多棵树,单独每棵树中节点的相对关系确定),每个节点对应的 v[] 值记录他与根节点的关系( 0:同类; 1:根吃他; 2:他吃根 )。当不在同一棵树上时,只需把 aa,bb 两个根节点连接,并修改变成aa子结点的bb相对于aa的关系。这里 根据 a ,b分别对应于aa ,bb的关系(即:v[a] ,v[b])就可以确定 bb相对于新根aa的关系。v[bb] = (v[a]-v[b]+3+d)%3。v[b]代表b作为子结点与bb的关系,反过来bb作为子结点与b的关系则为-v[b],加3是为了防止出现负数。
import java.util.*; import java.math.BigInteger; public class Main { static int father[]=new int[50005],v[]= new int[50005]; public static int find(int x) { int y; if(x!=father[x]) { y=father[x]; father[x]=find(father[x]); v[x]=(v[x]+v[y])%3; } else return x; return father[x]; } public static void link(int x,int y,int d) { int xx=find(x),yy=find(y); father[yy]=xx; v[yy]=(v[x]-v[y]+d+3)%3; } public static void main(String[] args) { Scanner in=new Scanner(System.in); int n,i,m,x,y,d,ans=0; Arrays.fill(v, 0); for(i=0;i<50005;i++)father[i]=i; n=in.nextInt();m=in.nextInt(); while(m--!=0) { d=in.nextInt(); x=in.nextInt(); y=in.nextInt(); if(x>n||y>n) ans++; else if(d==1) { if(find(x)!=find(y)) link(x,y,0); else if(v[x]!=v[y]) ans++; } else { if(find(x)!=find(y)) link(x,y,1); else if((v[x]+1)%3!=v[y]) ans++; } } System.out.println(ans); } }