洛谷 P4186

首先对于一棵子树,肯定是放在这棵子树中深度最小的叶节点。
那如何分析子树中深度最小的叶节点深度够不够小呢?
我们看到,我们关注叶节点深度是为了看它能不能在 Bessie 从根节点到达该子树的根之前(或同时)到达子树的根。
也就是说,最浅叶节点 到 子树根 的距离应该小于等于 树根 到 子树根 的距离。只要满足了这一点,该子树就可以只放一个农民。否则我们就要继续递归去考虑这个子树的各个子树。
DFS 求出每个节点的深度以及以它为根的子树的最浅叶节点的深度。

Code:

#include <bits/stdc++.h>
using namespace std;
const int N = 100005;
int n, k, ans;
int head[N], ver[N*2], nxt[N*2], cnt;
int dep[N], mn[N];

void add(int u, int v) {
	ver[++cnt] = v, nxt[cnt] = head[u], head[u] = cnt;
}

void dfs(int u, int fa) {
	int t = dep[u];
	for (int i = head[u]; i; i = nxt[i]) {
		int v = ver[i];
		if (v == fa) continue;
		dep[v] = dep[u] + 1, dfs(v, u);
		if (t == dep[u]) t = mn[v];
		else t = min(t, mn[v]); 
	}
	mn[u] = t;
}

void dfs1(int u, int fa) {
	if (mn[u] <= dep[u] * 2) return ++ans, void();
	for (int i = head[u]; i; i = nxt[i]) {
		int v = ver[i];
		if (v == fa) continue;
		dfs1(v, u); 
	}
}

int main() {
	scanf("%d%d", &n, &k);
	for (int i = 1, u, v; i < n; ++i) scanf("%d%d", &u, &v), add(u, v), add(v, u);
	dfs(k, 0), dfs1(k, 0);
	printf("%d", ans);
	return 0;
}
posted @ 2022-10-04 18:15  Kobe303  阅读(24)  评论(0编辑  收藏  举报