比赛 2025.12.13 评测结果 AAWAWAAWWAWWWAAAAWAWWAWAWWWWWW
题目名称 巡逻 最终得分 43
用户昵称 梦那边的美好BP 运行时间 0.870 s
代码语言 C++ 内存使用 5.04 MiB
提交时间 2025-12-13 11:51:51
显示代码纯文本
#include <iostream>
using namespace std;
const int N = 4e6 + 6;
int head[N], nxt[N], to[N], vl[N], num = 0;
void add(int x, int y, int z) {
    num++;
    to[num] = y;
    vl[num] = z;
    nxt[num] = head[x];
    head[x] = num;
}
int d1[N], d2[N], d = -0x3f3f3f3f, f1[N], f2[N];
void dfs(int u, int fa) {
    d1[u] = d2[u] = 0;
    f1[u] = f2[u] = 0;
    for (int i = head[u]; i; i = nxt[i]) {
        int v = to[i];
        if (v == fa) continue;
        dfs(v, u);
        int t = d1[v] + vl[i];
        if (t > d1[u]) {
            d2[u] = d1[u];
            d1[u] = t;
            vl[f1[u]] = 1;
            f1[u] = i;
            vl[i] = -1;
        } else if (t > d2[u]) {
            d2[u] = t;
            vl[f2[u]] = 1;
            f2[u] = i;
            vl[i] = -1;
        }
        d = max(d, d1[u] + d2[u]);
    }
}
int main() {
    freopen("xunluo.in", "r", stdin);
    freopen("xunluo.out", "w", stdout);
    int n, k;
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        int u, v;
        cin >> u >> v;
        add(u, v, 1);
        add(v, u, 1);
    }
    dfs(1, 0);
    int ans1 = d;
    d = -0x3f3f3f3f;
    dfs(1, 0);
    int ans2 = d;
    // cout << ans1 << " " << ans2;
    if (k == 1) {
        cout << (n - 1) * 2 - ans1 + 1;
    } else {
        cout << (n - 1) * 2 - ans1 - ans2 + 2;
    }
    return 0;
}