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 = 100000 + 10;
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 = 1; i <= n; i++)
        {
            father[i] = i;
            num[i] = 1;
        }
        for(int i = 1; i < n; i++)
        {
            scanf("%d%d%d", &u,&v,&w);
            if(w == 0)
            {
                Union(u,v);
            }
        }
        int temp = find_father(1);
        int ans = num[temp];

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

    return 0;
}