题意
你需要算出每一个点的颜色,满足一条边如果是 2 2 2 的倍数就要连接它的两个顶点的颜色相同。
思路
考虑图论,先将编号为 1 1 1 的点设置为 1 1 1,然后开始遍历它的邻接点。要是边是 2 2 2 的倍数,就将它的值设为与编号为 1 1 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;
}