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[],a[];//标记根节点
 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=; i<t; i++)
         {
             clc(a,);
             scanf("%d",&n);
             for(int i=; i<=n; i++)
             {
                 pre[i]=i;
             }
             for(int i=; i<n; i++)
             {
                 scanf("%d%d%d",&x,&y,&z);
                 if(!z) join(x,y);
             }
             for(int i=; i<=n; i++)
             {
                 a[find(i)]++;
             }
             long long ans=;
             for(int i=; i<=n; i++)
             {
                 int p=a[find(i)];
                 if(p) ans^=p;
                 else ans^=;
             }
             printf("%I64d\n",ans);
         }
     }
     return ;
 }