D - Timofey and a tree

题意：
给定一棵树，每个点带有各自的颜色，问是否能找出一个点当做根节点，并且它的子树颜色都是相同的。
思路：每颗子树之间的边的两端都是相同的颜色，对不同颜色的答案和是没有贡献的，那么如果找到两端颜色不同的就让不同边数++，两端点都++，如果有个点他的val值等于不同边的总数，那么它一定可以当做根节点了，因为子树中没有点对对答案造成影响。
AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N=100010;
int h[N];
int c[N];
int kind;
vector<pair<int,int>>query;
int main()
{
	int n;
	cin >>n;
	for(int i=0;i<n-1;i++)
	{
		int a,b;
		cin>>a>>b;
		query.push_back({a,b});
	}
	for(int i=1;i<=n;i++)
	{
		cin>>c[i];
	}
	for(auto x:query)
	{
		int a=x.first;
		int b=x.second;
		if(c[a]!=c[b])
		{
			kind++;
			h[a]++;
			h[b]++;
		}
	}
	for(int i=1;i<=n;i++)
	{
		if(h[i]==kind)
		{
			cout<<"YES"<<endl;
			cout<<i<<endl;
			return 0;
		}
	}
	cout<<"NO"<<endl;
	return 0;
 }