https://atcoder.jp/contests/abc126/tasks/abc126_d
题意:已知一棵树,每个节点有黑白两种颜色可选择,要求颜色相同的2个点距离是偶数。求一种可行的涂色方法。
思路:简单的模拟。从一个点出发,与他相连的边长是奇数,那个点的颜色就与这个不一样,否则一样。dfs一下就行了。
#include<cstdio>
#include<string>
#include<cstring>
#include<utility>
#include<cmath>
#include<map>
#include<queue>
#include<set>
#include<algorithm>
#include<vector>
#include<iostream>
#define ll long long
#define pii pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define inf 0x7fffffff
using namespace std;
vector<pii> ve[100010];
int a[100010];
int dfs(int x,int p)
{
for(int i=0;i<ve[x].size();i++)
{
if(ve[x][i].fi!=p)
{
a[ve[x][i].fi]=a[x]^ve[x][i].se;
dfs(ve[x][i].fi,x);
}
}
}
int main()
{
int i,j,k,n,m,x,y,z;
scanf("%d",&n);
for(i=1;i<n;i++)
{
scanf("%d%d%d",&x,&y,&z);
ve[x].push_back(mp(y,z&1));
ve[y].push_back(mp(x,z&1));
}
a[1]=0;
dfs(1,-1);
for(i=1;i<=n;i++)
{
printf("%d\n",a[i]);
}
return 0;
}