bzoj 1603: [Usaco2008 Oct]打谷机（拆点并查集）

1603: [Usaco2008 Oct]打谷机

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 977   Solved: 749
[ Submit][ Status][ Discuss]

Description

Farmer John有一个过时的打谷机（收割小麦），它需要带子来带动。发动机驱动轮1总是顺时针旋转的，用来带动转轮2，转轮2来带动转轮3，等等。一共有n（2<=n<=1000）个转轮（n-1条带子）。上面的图解描述了转轮的两种连接方式，第一种方式使得两个轮子旋转的方向相同，第二种则相反。 给出一串带子的信息： *Si—驱动轮 *Di—被动轮 *Ci—连接的类型（0=直接连接，1=交叉连接） 不幸的是，列出的信息是随即的。 作为样例，考虑上面的图解，n=4，转轮1是驱动轮，可以得知最后转轮4是逆时针旋转的。

Input

*第一行：一个数n *第二行到第n行：每一行有三个被空格隔开的数：Si，Di，Ci

Output

*第一行：一个单独的数，表示第n个转轮的方向，0表示顺时针，1表示逆时针。

Sample Input

4

2 3 0

3 4 1

1 2 0

Sample Output

1

这题没图，只好根据样例推题意：

两个点直接连接则旋转方式相同，交叉连接旋转方式相反

因为有两种状态（顺时针和逆时针），并且这两个状态可以互转

那么可以将每个齿轮拆成两个点Ai和Bi

如果齿轮i和j直接连接

那么将Ai的祖先和Aj的祖先连在一起，将Bi的祖先和Bj的祖先连在一起

如果齿轮i和j交叉连接

那么将Ai的祖先和Bj的祖先连在一起，将Aj的祖先和Bi的祖先连在一起

最后只要看A1和An在同一集合里，还是A1和Bn在同一集合里

前者说明A1和An旋转方式相同，都是顺时针，后者相反

#include<stdio.h>
#include<string.h>
int ufs[2005];
int Find(int x)
{
	if(ufs[x]==0)
		return x;
	return ufs[x] = Find(ufs[x]);
}
int main(void)
{
	int n, i, t1, t2, x, y, t;
	while(scanf("%d", &n)!=EOF)
	{
		memset(ufs, 0, sizeof(ufs));
		for(i=1;i<=n-1;i++)
		{
			scanf("%d%d%d", &x, &y, &t);
			if(t==0)
			{
				t1 = Find(x), t2 = Find(y);  ufs[t1] = t2;
				t1 = Find(x+n), t2 = Find(y+n);  ufs[t1] = t2;
			}
			else
			{
				t1 = Find(x), t2 = Find(y+n);  ufs[t1] = t2;
				t1 = Find(x+n), t2 = Find(y);  ufs[t1] = t2;
			}
		}
		if(Find(1)==Find(n))
			printf("0\n");
		else
			printf("1\n");
	}
	return 0;
}