BZOJ 1603: [Usaco2008 Oct]打谷机|dfs

深搜一边即可

#include<set>
#include<map>
#include<ctime>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define T 5006
#define MX 1e9
using namespace std;
int f[T],head[T],lst[T],nxt[T],v[T];
int n,m,tot;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
void insert(int x,int y,int z)
{
    lst[++tot]=y; nxt[tot]=head[x]; v[tot]=z; head[x]=tot;
    lst[++tot]=x; nxt[tot]=head[y]; v[tot]=z; head[y]=tot;
}
void dfs(int x,int fa)
{
    for(int i=head[x];i;i=nxt[i])
        if(lst[i]!=fa)
        {
            f[lst[i]]=v[i]^f[x];
            dfs(lst[i],x);
        }
}
int main()
{
    n=sc();
    for(int i=1;i<n;i++)
    {
        int x=sc(),y=sc(),c=sc();
        insert(x,y,c);
    }
    f[1]=0;
    dfs(1,0);
    cout<<f[n];
    return 0;
}