BestCoder Round #68 (div.2) 1002 tree

题意：给你一个图，每条边权值0或1，问每个点周围最近的点有多少个？

思路：并查集找权值为0的点构成的连通块。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
#define clc(a,b) memset(a,b,sizeof(a))
using namespace std;
int pre[100010],a[100010];//标记根节点
int find(int x)//查找根节点
{
    int r=x;
    while(pre[r]!=r)//返回根节点r
        r=pre[r];
    int i=x,j;
    while(i!=r)//路径压缩
    {
        j=pre[i];//在改变上级之前用临时变量j记录下他的值
        pre[i]=r;//把上级改为根节点
        i=j;
    }
    return r;
}
void join(int x,int y)//判断x,y是否连通.如果已经连通,就不用管了;如果不连通,就把它们所在的连通分支合并起
{
    int fx=find(x),fy=find(y);
    if(fx!=fy) pre[fy]=fx;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int t,n,x,y,z;
    while(~scanf("%d",&t))
    {
        for(int i=0; i<t; i++)
        {
            clc(a,0);
            scanf("%d",&n);
            for(int i=1; i<=n; i++)
            {
                pre[i]=i;
            }
            for(int i=1; i<n; i++)
            {
                scanf("%d%d%d",&x,&y,&z);
                if(!z) join(x,y);
            }
            for(int i=1; i<=n; i++)
            {
                a[find(i)]++;
            }
            long long ans=0;
            for(int i=1; i<=n; i++)
            {
                int p=a[find(i)];
                if(p) ans^=p;
                else ans^=1;
            }
            printf("%I64d\n",ans);
        }
    }
    return 0;
}