Codeforces 763A Timofey and a tree（思维）

传送门

题意：给一棵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;
}