
题意:给你一个图,每条边权值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 ;
}