poj 1182食物链(并查集)

时间:2024-06-11 15:34:20

算法思路:把那些确定了相对关系的节点放在同一棵树里(可以同时存在多棵树,单独每棵树中节点的相对关系确定),每个节点对应的 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); }
}