UVA

https://odzkskevi.qnssl.com/8b9d8bb43198f28efe782047d56ab361?v=1501987995
 1 /*
 2 题目大意：一个树上有n个节点，n-1条边，每个节点上有一个颜色
 3 现要求选取其中一个节点为根，要求子树的颜色相同，问是否能够达到要求。
 4 题目分析：两个有共同边的节点，如果颜色不一样，那么肯定会有其中一个点选为根
 5 记录这个点的度+1，总的counts+1，最后遍历数组，找出度和counts的大小相同的点并输出
 6 如果没有即是不能达到要求。
 7 */
 8 #include <cstdio>
 9 #include <cstring>
10 using namespace std;
11 int const maxn =  100005;
12 struct node{
13     int u,v;
14 }a[maxn];
15 int c[maxn];
16 int degree[maxn];
17 int main(){
18     int n;
19     scanf("%d",&n);
20     for(int i=1;i<n;i++)
21         scanf("%d%d",&a[i].u ,&a[i].v );
22     for(int i=1;i<=n;i++) scanf("%d",&c[i]);
23     int counts=0;
24     memset(degree,0,sizeof(degree));
25     for(int i=1;i<n;i++){
26         if(c[a[i].u]!=c[a[i].v]){
27             counts++;
28             degree[a[i].u ]++;
29             degree[a[i].v ]++;
30         }
31     }
32     for(int i=1;i<=n;i++){
33         if(counts==degree[i]){
34             printf("YES
%d
",i);
35             return 0;
36         }
37     }
38     printf("NO
");
39     return 0;
40 }
41 /////Runtime error on test
42 #include <iostream>
43 #include <stdio.h>
44 #include <string.h>
45 using namespace std;
46 const int maxn=100005;
47 struct node {
48     int u,v;
49 }a[maxn];
50 int c[maxn],degree[maxn];
51 int main(){
52     int n;
53     scanf("%d",&n);
54     for(int i=1;i<n;i++)
55         scanf("%d%d",&a[i].u,&a[i].v);
56     for(int i=1;i<=n;i++)
57     scanf("%d",&c[i]);
58     int ans=0;
59     memset(degree,0,sizeof(degree));
60     for(int i=1;i<n;i++){
61         if(c[a[i].u]!=c[a[i].v]){
62             ans++;
63             degree[a[i].u]++;
64             degree[a[i].v]++;
65         }
66     }
67     for(int i=1;i<=n;i++){
68         if(ans==degree[i]){
69         printf("YES
%d
",i);
70         return 0;
71         }
72     }
73         printf("NO
");
74         return 0;
75 }