Codeforces Round 395 C Timofey and a tree 树

题意：一颗树，每个节点都有一个颜色值。如果能从中拿掉一个点后，所形成的深林中子树所有节点颜色相同，但不同子树颜色可以不同。

思路：把两端点颜色不同不同的线段选出来，然后记录边两端不同颜色出现的次数，某个点出边（不同颜色的边）的次数等于总的不同颜色的边数。那说明符合题意。

#include<cstdio>#include<queue>#include<iostream>#include<vector>#include<map>#include<cstring>#include<string>#include<set>#include<stack>#include<algorithm>#define cle(a) memset(a,0,sizeof(a))#define inf(a) memset(a,ox3f,sizeof(a))#define ll long long#define FOR(i,n) for(int i=0;i<n;i++)using namespace std;const int INF = ( 2e9 ) + 2;const int maxn = 1e5+10;int u[maxn],v[maxn],c[maxn];map<int,int> mp;int main(){int n;scanf("%d",&n);for(int i=0;i<n-1;i++)scanf("%d%d",&u[i],&v[i]);for(int i=1;i<=n;i++)scanf("%d",&c[i]);int cnt=0;for(int i=0;i<n-1;i++){if(c[u[i]]!=c[v[i]]){cnt++;mp[u[i]]++;mp[v[i]]++;}}int flag=0,ans=0;for(int i=1;i<=n;i++){if(mp[i]==cnt)flag=1,ans=i;}if(flag)printf("YES\n%d\n",ans);else printf("NO\n");}