树的直径的求法

一、树的直径

给定一颗树，树的每条边都有一个权值, 树中任意两点都有一条唯一的简单路径，路径长度为连接两点的路径上的边权之和，路径长度最长的一条为树的直径，往往说的直径既可以指路径长度，也可以指具体路径(即经过哪些点)。直径不唯一。

一般使用两种方法求直径：两次dfs/bfs、树形dp，时间复杂度都为O（n）。

本文主要关注两次dfs

二、两次dfs

思路：由于树上任意两点都有唯一的路径，用p、q表示直径的两个端点。

从直径一端p出发进行dfs/bfs，dist[i]表示p到i的路径长度，那么dist最大的点就是q，此时dist[q]就是直径长度，所以已知p就可以求出q。

如何求p？从任意点出发，找到距离出发点最远的节点，该节点就是p。（第一次dfs）

代码：

#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int maxn = 1e5 + 5;
vector<int> e[maxn];
int n, dist[maxn], pre[maxn]; // pre:前驱(父)节点
void dfs(int u, int fa, int &p) {
    pre[u] = fa;
    if (dist[u] > dist[p])
        p = u;
    for (int v : e[u]) {
        if (v == fa)
            continue;
        dist[v] = dist[u] + 1;
        dfs(v, u, p);
    }
}
// 输出具体路径, pre[p] == 0
void print(int x) {
    if (x == 0)
        return;
    print(pre[x]);
    cout << x << ' ';
}
int main() {
    cin >> n;
    for (int i = 0; i < n - 1; ++i) {
        int u, v;
        cin >> u >> v;
        e[u].push_back(v);
        e[v].push_back(u);
    }
    int p, q;
    dist[1] = 0;
    dfs(1, 0, p = 1);
    dist[p] = 0;
    dfs(p, 0, q = p);
    print(q);
    cout << '\n' << dist[q] << '\n';
    return 0;
}