| 比赛 |
2025.12.13 |
评测结果 |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA |
| 题目名称 |
巡逻 |
最终得分 |
100 |
| 用户昵称 |
LikableP |
运行时间 |
0.361 s |
| 代码语言 |
C++ |
内存使用 |
2.73 MiB |
| 提交时间 |
2025-12-13 11:12:52 |
显示代码纯文本
#include <cstdio>
#include <cctype>
template <typename T> T read() {
T res = 0, f = 1;
char ch = getchar();
for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1;
for (; isdigit(ch); ch = getchar()) res = (res << 3) + (res << 1) + (ch ^ 48);
return res * f;
}
void write(__int128 x, char ed = '\n') {
if (x < 0) x = -x, putchar('-');
static int sta[64], top = 0;
do {
sta[++top] = x % 10;
x /= 10;
} while (x);
while (top) {
putchar(sta[top--] ^ 48);
}
putchar(ed);
}
template <typename T> void write(T x, char ed = '\n') {
write((__int128)x, ed);
}
#include <algorithm>
const int MAXN = 2e6 + 10;
struct EDGE {
int v, w, next;
} edge[MAXN << 1];
int head[MAXN], edgeNum = 1;
void AddEdge(int u, int v, int w) {
edge[++edgeNum] = {v, w, head[u]};
head[u] = edgeNum;
}
int depth[MAXN], far, fa[MAXN];
void dfs1(int u, int fa) {
::fa[u] = fa;
for (int i = head[u]; i; i = edge[i].next) {
int v = edge[i].v;
if (v == fa) continue;
depth[v] = depth[u] + 1;
if (depth[v] > depth[far]) far = v;
dfs1(v, u);
}
}
int dp[MAXN], d;
void dfs2(int u, int fa) {
for (int i = head[u]; i; i = edge[i].next) {
int v = edge[i].v, w = edge[i].w;
if (v == fa) continue;
dfs2(v, u);
d = std::max(d, dp[u] + dp[v] + w);
dp[u] = std::max(dp[u], dp[v] + w);
}
}
int n, k;
int L1, L2;
int main(int argc, char **argv) {
#ifdef LOCAL
freopen("!input.in", "r", stdin);
freopen("!output.out", "w", stdout);
#elif defined(FS)
freopen(argv[1], "r", stdin);
freopen(argv[2], "w", stdout);
#else
freopen("xunluo.in", "r", stdin);
freopen("xunluo.out", "w", stdout);
#endif
n = read<int>(), k = read<int>();
for (int i = 1; i <= n - 1; ++i) {
int u = read<int>(), v = read<int>();
AddEdge(u, v, 1);
AddEdge(v, u, 1);
}
dfs1(1, 0);
depth[far] = 0;
dfs1(far, 0);
L1 = depth[far];
while (far) {
for (int i = head[far]; i; i = edge[i].next) {
if (edge[i].v == fa[far]) {
edge[i].w = -1;
edge[i ^ 1].w = -1;
break;
}
}
far = fa[far];
}
dfs2(1, 0);
L2 = d;
write(2 * n - L1 - (k == 2 ? L2 : 1));
return 0;
}