Codeforces Round #395 (Div. 1), problem: (A) Timofey and a tree 【思维】

题意

是否存在一个点，去掉之后子图的颜色一样，如果有，就输出一个，如果有多个，就输出任意一个

思路

我们把这个题的边和点分开考虑，如果一个图中有不同的颜色，则必有一条边的两端不是同一种颜色，我们称这种边为特殊边，特殊边上的两个点称为特殊点，我们只需要比较特殊边的数量和特殊点被计算几次就可以了，如果特殊边和特殊点被计算的次数相同，则这就是我们需要找的点。

举个栗子，3，1 2，2 3，1 2 1这组数据，特殊边为1 2，2 3，数量为2，而1被计算了1次,2被计算了两次，3被计算了1次，所以2就是我们需要找的点，脑中模拟一下这个图就明白了

code

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
const int maxn=1e5+5;
int a[maxn],b[maxn];
int c[maxn],p[maxn];
int n;
int main(){
    cin>>n;
    for(int i=0;i<n-1;i++)
        cin>>a[i]>>b[i];
    for(int i=1;i<=n;i++)
        cin>>c[i];
    int ans=0;
    for(int i=0;i<n-1;i++){
        if(c[a[i]]!=c[b[i]])
            ++ans,p[a[i]]++,p[b[i]]++;
    }
    for(int i=1;i<=n;i++){
        if(p[i]==ans){
            cout<<"YES"<<endl;
            cout<<i<<endl;
            return 0;
        }
    }
    cout<<"NO"<<endl;
    return 0;
}

学如逆水行舟，不进则退