Codeforce 1068：E. Multihedgehog（dfs，图，分析）

题目大意：定义1 - 刺猬图：存在一个点度 >= 3，其它点度为 1.定义一个 k - 刺猬图：由一个中心连接至少3个 (k - 1) - 刺猬图。输入 n,k，输入一个 n 个点的树，判断它是不是一个 k - 刺猬图。

---

题解：首先刺猬图是一棵树（题目有说），将刺猬图画出来会发现，如果你准确的找到了一个刺猬图的根，它满足：每个叶子结点到这个根结点的距离都为 这个刺猬图的阶数，每个子树都是一个 更低一阶的刺猬图。
换句话说：如果一个图是 k 阶刺猬图，所有叶子结点到根结点的距离为 k，且除叶子结点外每个结点的儿子结点个数 $\ge 3$，重点在于怎么找 根。

如果一个树是 k 阶刺猬图，那么它的直径为 2 * k，且根节点一定在直径的中心，通过两次 dfs 可以找出直径，然后回溯找到直径的中间点（也就是根结点），从根节点开始dfs遍历一次，判断每个结点是否满足上面两条性质，复杂度$O(n)$

---

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 4e5 + 10;
int fa[maxn];
int head[maxn],nxt[maxn],to[maxn],cnt;
int d[maxn],son[maxn];
int n,k;
void init() {
	cnt = 0;
	fill(head,head + n + 1,-1);
	fill(d,d + n + 1,0);
}
void add(int u,int v) {
	to[cnt] = v;
	nxt[cnt] = head[u];
	head[u] = cnt++;
}
int dfs(int s,int f) {
	int ans = s;
	fa[s] = f;
	for(int i = head[s]; i + 1; i = nxt[i])	{
		if(to[i] == f) continue;
		d[to[i]] = d[s] + 1;
		int p = dfs(to[i],s);
		if(d[p] > d[ans]) ans = p; 
	}
	return ans;
}
void dfs2(int s,int f) {
	for(int i = head[s]; i + 1; i = nxt[i]) {
		if(to[i] == f) continue;
		d[to[i]] = d[s] + 1;
		son[s]++;
		dfs2(to[i],s);
	}
}
int main() {
	scanf("%d%d",&n,&k);
	init();
	int u,v;
	for(int i = 1; i < n; i++) {
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	d[0] = -1;
	d[1] = 0;
	int p = dfs(1,0);
	d[p] = 0;
	int rt = dfs(p,0);
	if(d[rt] != 2 * k) puts("NO");
	else {
		for(int i = 1; i <= k; i++)
			rt = fa[rt];
		d[rt] = 0;
		dfs2(rt,0);
		bool f = true;
		for(int i = 1; i <= n; i++) {
			if(son[i] == 0 && d[i] != k) {
				f = false;break;
			}
			else if(son[i] > 0 && son[i] < 3) {
				f = false;break;
			}
		}
		if(f) puts("Yes");
		else puts("No");
	}
	return 0;
}