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=,f=; char ch=getchar();
     while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}
     while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}
     return x*f;
 }
 int head[N],cnt;
 struct node{
     int next,to;
 }e[N<<];
 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=; i<n; i++)
     {
         int x=ra(),y=ra();
         insert(x,y);
         insert(y,x);
     }
     for (int i=; i<=n; i++)
         col[i]=ra();
     dfs(,);
     for (int i=; i<=n; i++)
     {
         int sum=;
         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 ;
         }
     }
     cout<<"NO";
     return ;
 }