题意： 给定一棵树，每个点带有各自的颜色，问是否能找出一个点当做根节点，并且它的子树颜色都是相同的。 思路：每颗子树之间的边的两端都是相同的颜色，对不同颜色的答案和是没有贡献的，那么如果找到两端颜色不同的就让不同边数++，两端点都++，如果有个点他的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; }