| 比赛 |
2025.12.13 |
评测结果 |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA |
| 题目名称 |
巡逻 |
最终得分 |
100 |
| 用户昵称 |
淮淮清子 |
运行时间 |
0.437 s |
| 代码语言 |
C++ |
内存使用 |
5.23 MiB |
| 提交时间 |
2025-12-13 10:32:53 |
显示代码纯文本
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN = 1e5 + 5;
struct node{
int to, nxt, len;
}e[MAXN * 2];
int tot = 0, h[MAXN];
int d[MAXN];
bool vis[MAXN];
void add(int x, int y, int len){
e[++ tot] = {y, h[x], len};
h[x] = tot;
}
int n, k, d1 = 1, d2 = 0, d3 = 1, d4 = 0;
int f[MAXN];
int dp[MAXN], maxx;
void dfs(int u, int fa){
dp[u] = 0;
for(int i = h[u]; i; i = e[i].nxt){
int v = e[i].to;
if(v == fa) continue;
e[i].len = (vis[u] && vis[v]) ? -1 : 1;
dfs(v, u);
maxx = max(maxx, dp[u] + dp[v] + e[i].len);
dp[u] = max(dp[u], dp[v] + e[i].len);
}
}
void dfs1(int u, int fa){
for(int i = h[u];i;i = e[i].nxt){
int v = e[i].to;
if(v == fa) continue;
d[v] = d[u] + e[i].len;
if(d[v] > d[d1]){
d1 = v;
}
dfs1(v, u);
}
}
void dfs2(int u, int fa){
f[u] = fa;
for(int i = h[u];i;i = e[i].nxt){
int v = e[i].to;
if(v == fa) continue;
d[v] = d[u] + e[i].len;
if(d[v] > d[d2]){
d2 = v;
}
dfs2(v, u);
}
}
int main(){
freopen("xunluo.in", "r", stdin);
freopen("xunluo.out", "w", stdout);
ios::sync_with_stdio(0);
cin.tie(0);
cin >> n >> k;
for(int i = 1;i < n;i ++){
int a, b; cin >> a >> b;
add(a, b, 1), add(b, a, 1);
}
dfs1(1, 0);
memset(d, 0, sizeof(d));
dfs2(d1, 0);
int l1 = d[d2];
if(k == 1){
cout << 2 * (n - 1) - l1 + 1 << '\n';
return 0;
}
memset(vis, 0, sizeof(vis));
int tmp = d2;
while(tmp != 0){
vis[tmp] = true;
tmp = f[tmp];
}
maxx = 0;
dfs(1, 0);
int l2 = maxx;
cout << 2 * (n - 1) - (l1 - 1) - (l2 - 1) << '\n';
return 0;
}