题意:一棵树,权值只有0和1,找到每个点与之相距最近的点的个数, (包括这个点自己,也就是说,等价于找每个点与之相距为0的点的个数)。刚开始一直迷惑为什么不考虑一个点与相连的点权都为1的情况,原来点是可以与本身相连距离为0,所以如果点与相连的点距离都为1,则无需考虑,肯定只能与自己相连;
思路:运用并查集,找出点与点距离为0构成的块,并记录根节点的连通数目(包括自己);
代码:
#include<stdio.h>
int a[100005];
int vag[100005];
int find(int tt)
{
if(tt!=a[tt])
a[tt]=find(a[tt]);
return a[tt];
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int i,j,k,n;
scanf("%d",&n);
for(i=1;i<=n;i++)
{
a[i]=i;
vag[i]=1;
}
for(i=1;i<n;i++)
{
int zx,zy,zz;
scanf("%d%d%d",&zx,&zy,&zz);
int dx=find(zx);
int dy=find(zy);
if(!zz)
{
a[dx]=dy;
vag[dy]=vag[dy]+vag[dx];
}
}
int ans=0;
for(i=1;i<=n;i++)
{
int ax=find(i);
ans=ans^(vag[ax]);
}
printf("%d\n",ans);
}
}