题意:给一棵n个点的无根树,每个点有一个颜色,问是否存在一个点使得以它为根时,每棵子树内颜色相同。
题解:第一反应树形dp,想了想好像没法搞......树上分块这些好像也不行。看了一眼题解...这个想法真的是...一针见血...如果所有点颜色相同那输出YES再随便输个点就好。如果有颜色不同的点对,那么找一条两个端点颜色不同的边,如果他们两个都不能作为合法的根,那么答案是NO(别的任何一点为根,这条边所在的子树不满足条件)。否则答案为YES。
复杂度为级别,真是......不知道该说啥...刁钻?
类似题目:https://codeforces.com/problemset/problem/1169/B
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e5+4;
int n;
struct Edge {
int v,nxt;
}e[N<<1];
int head[N],etot;
int c[N];
inline int read() {
int x=0,f=1;char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x*f;
}
inline void adde(int u,int v) {
e[++etot].nxt=head[u],e[etot].v=v,head[u]=etot;
}
inline bool dfs(int p,int fa,int mk) {
if (c[p]^mk) return false;
for (int i=head[p];~i;i=e[i].nxt) {
int v=e[i].v;
if (v^fa&&!dfs(v,p,mk))
return false;
}
return true;
}
inline bool work(int p) {
for (int i=head[p];~i;i=e[i].nxt) {
int v=e[i].v;
if (!dfs(v,p,c[v]))
return false;
}
return true;
}
int ans=-1;
int main() {
memset(head,-1,sizeof(head));
n=read();
for (register int i=1;i<n;++i) {
int u=read(),v=read();
adde(u,v);
adde(v,u);
}
for (register int i=1;i<=n;++i) c[i]=read();
for (register int x=1;x<=n;++x)
for (int i=head[x];~i;i=e[i].nxt) {
int y=e[i].v;
if (c[x]^c[y]) {
bool a=work(x);
bool b=work(y);
if (a) {
puts("YES");
printf("%d\n",x);
return 0;
} else if (b) {
puts("YES");
printf("%d\n",y);
return 0;
} else {
puts("NO");
return 0;
}
}
}
puts("YES");
puts("1");
return 0;
}