BC68(HD5606) 并查集+求集合元素

tree

 Accepts: 143

 Submissions: 807

 Time Limit: 2000/1000 MS (Java/Others)

 Memory Limit: 65536/65536 K (Java/Others)

问题描述

有一个树(nn个点, n-1n−1条边的联通图),点标号从11~nn,树的边权是00或11.求离每个点最近的点个数(包括自己).

输入描述

第一行一个数字TT,表示TT组数据.
对于每组数据,第一行是一个nn,表示点个数,接下来n-1n−1,每行三个整数u,v,wu,v,w,表示一条边连接的两个点和边权.
T = 50,1 \leq n \leq 100000, 1 \leq u,v \leq n,0 \leq w \leq 1T=50,1≤n≤100000,1≤u,v≤n,0≤w≤1.

输出描述

对于每组数据,输出答案.
考虑到输出规模过大,设ans_ians​i​​表示第ii个点的答案.你只需输出ans_1 \ xor \ ans_2 \ xor \ ans_3.. \ xor \ ans_nans​1​​ xor ans​2​​ xor ans​3​​.. xor ans​n​​即可.

输入样例

1
3
1 2 0
2 3 1

输出样例

1

Hint

ans_1 = 2ans​1​​=2
ans_2 = 2ans​2​​=2
ans_3 = 1ans​3​​=1
2 \ xor \ 2 \ xor \ 1=12 xor 2 xor 1=1, 因此输出11.

马丹！！我想骂人！刚开始做这道题就开始开了一个数组求每一个为0的两个点的数量，最后猛然发现如果三个数都是0答案就不对，然后有想到了深搜，用vector,存与之相连为0的点，写到一半，发现又不对！马丹，猛然想起就是并查集啊，把为0的点放在一起，就是就每个集合中数的个数嘛！敲了不到10分钟敲完，卡在8：45提交了，WA了，擦！猛然发现一个细节给忽略了，果断改了，8：46了。。。呵呵呵呵。。刚才提交a了，擦！太弱了，伤心的是又掉rating了....

 #include <iostream>
 #include <cstring>
 #include <algorithm>
 #include <cstdio>
 #include <vector>
 using namespace std;
 const int MAX =  + ;
 int num[MAX],father[MAX];
 int find_father(int x)
 {
     int r = x;
     while(r != father[r])
         r = father[r];
     int i = x,j;
     while(r != father[i])
     {
         j = father[i];
         father[i] = r;
         //num[r] += num[father[i]] 擦！就是这！没必要啊，以为此时num[r]中肯定含有num[father[i]]，因为这只是跟新直系跟节点，数量并不需修改！！！
         i = j;
     }
     return r;
 }
 void Union(int x, int y)
 {
     int fx = find_father(x);
     int fy = find_father(y);
     if(fx != fy)
     {
         father[fx] = fy;
         num[fy] += num[fx];
     }
 }
 int main()
 {
     int t;
     scanf("%d", &t);
     while(t--)
     {
         int u,v,w,n;
         scanf("%d", &n);
         for(int i = ; i <= n; i++)
         {
             father[i] = i;
             num[i] = ;
         }
         for(int i = ; i < n; i++)
         {
             scanf("%d%d%d", &u,&v,&w);
             if(w == )
             {
                 Union(u,v);
             }
         }
         int temp = find_father();
         int ans = num[temp];

         for(int i = ; i <= n; i++)
         {
             int fx = find_father(i);
             ans = ans ^ num[fx];
         }
         printf("%d\n",ans);
     }

     return ;
 }