poj 1182食物链(并查集)

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


原文地址:https://www.cnblogs.com/pangblog/p/3317983.html