cf 763A. Timofey and a tree

呵呵呵，直接判断是不是一个点连起来所有的特殊边（连接2不同颜色的点的边）

（一开始还想各种各样奇怪的dfs。。。垃圾）

#include<bits/stdc++.h> 
#define LL long long 
#define N 100005
#define lowbit(x) x&(-x)
using namespace std;
inline int ra()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
    while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
int head[N],cnt;
struct node{
    int next,to;
}e[N<<1];
int n,col[N],tot;
void insert(int x, int y)
{
    e[++cnt].next=head[x];
    e[cnt].to=y;
    head[x]=cnt;
}
void dfs(int x, int fa)
{
    for (int i=head[x];i;i=e[i].next)
    {
        if (e[i].to==fa) continue;
        if (col[e[i].to]!=col[x]) tot++;
        dfs(e[i].to,x); 
    }
}
int main()
{
    int n=ra();
    for (int i=1; i<n; i++)
    {
        int x=ra(),y=ra();
        insert(x,y);
        insert(y,x);
    }
    for (int i=1; i<=n; i++)
        col[i]=ra();
    dfs(1,0);
    for (int i=1; i<=n; i++)
    {
        int sum=0;
        for (int j=head[i];j;j=e[j].next)
            if (col[e[j].to]!=col[i]) sum++;
        if (sum==tot) 
        {
            cout<<"YES"<<endl<<i;
            return 0;
        }
    }
    cout<<"NO";
    return 0;
}