2021-11-04

最新推荐文章于 2024-10-30 16:52:01 发布

做一只大熊猫

最新推荐文章于 2024-10-30 16:52:01 发布

阅读量98

点赞数 1

分类专栏： # 树的直径文章标签：图论

本文链接：https://blog.csdn.net/qq_34261446/article/details/121136389

版权

树的直径专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题意

给我们一颗 n 个节点之间的树，要求我们在这棵树上选择 2 个不同的节点，使其他节点到这 2 个节点中的一个的距离尽可能的小，输出所有点的最小最大值。

思路

我们在树的直径上二分二分一个距离。在直径上找到两个对称点，这两个对称点到对应的叶子节点的距离为我们二分的距离：md。
然后对这两个对称点出发，暴力跑深度为 md 的距离，并标记走过的点，最后判读所有的点是否都被标记。

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 2e5 + 10;
int n;
struct Edge
{
    int v, ne;
} e[N * 2];
int h[N], tot;
int dep[N], pre[N], path[N], len;
void add(int u, int v)
{
    e[++ tot] = { v, h[u] }; h[u] = tot;
}

void init(int n)
{
    for (int i = 0; i <= n; i ++) h[i] = 0, pre[i] = 0, dep[i] = 0;
    tot = 0; 
    len = 0;
}


void dfs(int u, int p, int deep, int & d, int flag)     //处理树的重心和路径
{
    dep[u] = deep;
    if (deep > dep[d]) d = u;
    for (int i = h[u]; i; i = e[i].ne) {
        int v = e[i].v;
        if (v == p) continue;
        if (flag) pre[v] = u;
        dfs(v, u, deep + 1, d, flag);
    }
}

void sov_path()
{
    int s = 0, t = 0;
    dfs(1, 0, 0, s, 0);
    for (int i = 0; i <= n; i ++) dep[i] = 0;
    dfs(s, 0, 0, t, 1);
    while (t)
    {
        path[++ len] = t;
        t = pre[t];
    }
}

bool vis[N];
void dfs(int u, int p, int deep, int md)
{
    if (deep > md) return;
    vis[u] = 1;
    for (int i = h[u]; i; i = e[i].ne) {
        int v = e[i].v;
        if (v == p) continue;
        dfs(v, u, deep + 1, md);
    }
}

bool check(int md)
{
    for (int i = 1; i <= n; i ++) vis[i] = 0;
    dfs(path[md + 1], 0, 0, md);
    dfs(path[len - md], 0, 0, md);
    for (int i = 1; i <= n; i ++) if (! vis[i]) return false;
    return true;
}


int main()
{
    int T; scanf("%d", &T);
    while (T --)
    {
        scanf("%d", &n);
        init(n);
        int u, v;
        for (int i = 1; i < n; i ++) {
            scanf("%d %d", &u, &v);
            add(u, v);
            add(v, u);
        }

        sov_path();
        int x, y;
        int l = 0, r = (len - 1) / 2, ans = 0;
        while (l <= r)
        {
            int md = (l + r) >> 1;
            if (check(md)) r = md - 1, ans = md, x = path[md + 1], y = path[len - md];
            else l = md + 1;
        }

        if (x == y) y = pre[x];
        printf("%d %d %d\n", ans, x, y);
    }

    return 0;
}