bestcoder#68 div2tree【并查集】

tree

 

 Accepts: 152

 

 Submissions: 697

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

 

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

问题描述

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

输入描述

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

输出描述

对于每组数据,输出答案.
考虑到输出规模过大,设ans_ians​i​​表示第$i$个点的答案.你只需输出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
$2xor2xor1=1$, 因此输出$1$.

貌似没结束就发题解有点张扬啊 低调低调~~1A飘过

并查集再维护一个记录集合元素个数的数组，每次合并的时候更新就好啦 啦啦啦

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,t,a[100005],num[100005],cnt;
int fnd(int x)
{
    if(x!=a[x]) a[x]=fnd(a[x]);
    return a[x];
}
void addto(int x,int y)
{
    x=fnd(x),y=fnd(y);
    if(x==y) {
        return ;
    }
    a[y]=a[x];
    num[x]+=num[y];
    return ;
}
int main()
{
  //  freopen("cin.txt","r",stdin);
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        cnt=0;
        for(int i=1;i<=n;i++) {num[i]=1;a[i]=i;}
        for(int i=1;i<n;i++)
        {
            int u,v,q;
            scanf("%d%d%d",&u,&v,&q);
            if(q==0)  addto(u,v);
        }
        //cout<<"44544"<<endl;
        cnt=0;
        for(int i=1;i<=n;i++)
        {
            if(a[i]==i&&(num[i]&1)) cnt^=num[i];
        }
        printf("%d\n",cnt);
    }
    return 0;
}