图论-树的直径

图论-树的直径

问题

我们将一颗树$T=(V,E)$的直径定义为${\max }_{u,v\in V}\delta (u,v)$，其中$\delta (u,v)$代表从u到v的最短路径。也就是树中所有最短路径的最大值。请设计一个算法，求这颗树的直径。

两次BFS或BFS

我们使用两次BFS即可，第一次随机选取一个节点当做为源节点，之后找到里源节点最远的那个节点，就是直径的一个端点。

直径要么经过源节点S，要么不经过，我们可以证明，其中一个个端点在BFS的最外层上。简单证明：使用反证法，如果直径的端点v不在最外层上，那么其必然会存在一条路径从v到最外层，那么就会存在更远的直径。

之后，我们再以这个节点作为源节点，找到最远节点，之间的距离就是最远节点。

树形DP

树的直径属于基本的树数上路径DP问题。

我们定义toLeaf(x)为节点x到任意一个子树下的叶子节点的最大距离。

#include <bits/stdc++.h>

#define FR freopen("in.txt", "r", stdin)

typedef long long ll;
using namespace std;

struct Edge
{
    int to;
    int nxt;
} e[200005];

int head[100005];

int toLeaf[100005];

int tot = 0;

void add(int u, int v)
{
    tot++;
    e[tot].to = v;
    e[tot].nxt = head[u];
    head[u] = tot;
}

int n;

int ans = 0;

void dfs(int u, int r)
{
    int mx = 0;
    for (int adj = head[u]; adj; adj = e[adj].nxt)
    {
        int to = e[adj].to;

        if (to == r)
            continue;
        dfs(to, u);
        toLeaf[u] = max(toLeaf[u], toLeaf[to] + 1);
        ans = max(ans, mx + toLeaf[u]);
        mx = max(mx, toLeaf[u]);
    }
}

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

    dfs(1, 0);

    printf("%d", ans);
    return 0;
}

带有负权边（点）的求解

两遍DFS或者BFS都失效了，只能使用树形DP。