洛谷 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;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话