[ABC126D] Even Relation 题解

题意

你需要算出每一个点的颜色，满足一条边如果是 $2$ 的倍数就要连接它的两个顶点的颜色相同。

思路

考虑图论，先将编号为 $1$ 的点设置为 $1$，然后开始遍历它的邻接点。要是边是 $2$ 的倍数，就将它的值设为与编号为 $1$ 的点相同，否则取反。接着，在将它继续遍历，直到遍历完为止。

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n;
struct op{
	int from,to,sum;//链式前项星存图
}a[200005];
int cnt=1;
int head[200005];
void hb(int x,int y,int sum)//合并
{
	a[cnt].from=head[x];
	a[cnt].to=y;
	a[cnt].sum=sum;
	head[x]=cnt++;
}
int ans[200005];
bool bj[200005];
void dfs(int x)//搜索
{
	for(int i=head[x];i!=0;i=a[i].from)//遍历邻接点
	{
		int y=a[i].to;//求出邻接点
		if(bj[y]==1)//如果遍历过
		{
			continue;
		}
		if(a[i].sum%2==0)//是二的倍数
		{
			bj[y]=1;//标记
			ans[y]=ans[x];//设为相同
			dfs(y);//继续搜索
		}
		else
		{
			bj[y]=1;//标记
			if(ans[x]==1)//设为相反
			{
				ans[y]=0;
			}
			else
			{
				ans[y]=1;
			}
			dfs(y);//继续搜索
		}
	}
}
signed main()
{
	cin>>n;
	int x,y,z;
	for(int i=1;i<n;i++)
	{
		scanf("%lld%lld%lld",&x,&y,&z);
		hb(x,y,z);//合并x到y
		hb(y,x,z);//合并y到x
	}
	ans[1]=1;//将颜色设为1
	bj[1]=1;//标记已经遍历过
	dfs(1);//搜索
	for(int i=1;i<=n;i++)
	{
		printf("%lld\n",ans[i]);//输出答案
	}
	return 0;
}