[BZOJ3124][SDOI2013]直径（DFS）

先DP求一遍直径。下面记$f[u]$和$g[u]$分别为自节点$u$向下延伸到叶子节点的最长链和次长链的长度，并把所有满足「$f[u]+g[u]$的值等于该树直径」的节点$u$标记为关键点。
首先给出一个性质：不存在深度相同的关键点，并且深度小的关键点是深度大的关键点的祖先。
证明：如果有两个关键点不满足上述限制，那么这两个关键点对应的两条直径一定不重叠。此时就一定存在一条比直径更长的链。
从上面的证明还可得到：深度最大的关键点被所有的直径经过。设深度最大的关键点为$X$。
这时候就可以简单判断情况后DFS了。在DFS的过程中，记当前达到的节点为$u$，则：
1、对于$u$的子节点$v$，如果有且仅有一个$v$使得$f[u]=f[v]+val(u,v)$（$val(u,v)$为边$(u,v)$的权值），那么$(u,v)$被所有的直径经过。然后跳到$v$继续DFS。
2、对于$u$的子节点$v$，如果满足$f[u]=f[v]+val(u,v)$的点$v$不止一个，那么DFS结束。$u$的子树内任意一条边都不在所有的直径上。
代码：

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
inline int read() {
    int res = 0; bool bo = 0; char c;
    while (((c = getchar()) < '0' || c > '9') && c != '-');
    if (c == '-') bo = 1; else res = c - 48;
    while ((c = getchar()) >= '0' && c <= '9')
        res = (res << 3) + (res << 1) + (c - 48);
    return bo ? ~res + 1 : res;
}
const int N = 2e5 + 5;
typedef long long ll;
int n, ecnt, nxt[N << 1], adj[N], go[N << 1], val[N << 1], Ans, dep[N];
ll f[N], g[N];
void add_edge(int u, int v, int w) {
    nxt[++ecnt] = adj[u]; adj[u] = ecnt; go[ecnt] = v; val[ecnt] = w;
    nxt[++ecnt] = adj[v]; adj[v] = ecnt; go[ecnt] = u; val[ecnt] = w;
}
void dfs(int u, int fu) {
    dep[u] = dep[fu] + 1;
    for (int e = adj[u], v; e; e = nxt[e]) {
        if ((v = go[e]) == fu) continue; dfs(v, u);
        if (f[v] + val[e] > f[u]) g[u] = f[u], f[u] = f[v] + val[e];
        else if (f[v] + val[e] > g[u]) g[u] = f[v] + val[e];
    }
}
void dfs2(int u, int fu) {
    int tot = 0, w = -1;
    for (int e = adj[u], v; e; e = nxt[e]) {
        if ((v = go[e]) == fu) continue;
        if (f[u] == f[v] + val[e]) tot++, w = v;
    }
    if (tot >= 2) return;
    if (w != -1) Ans++, dfs2(w, u);
}
void solve() {
    int i, cnt1 = 0, cnt2 = 0, top = 0, u1, u2, v, tt = 0;
    dfs(1, 0); ll ans = 0;
    for (i = 1; i <= n; i++) ans = max(ans, f[i] + g[i]);
    printf("%lld\n", ans);
    for (i = 1; i <= n; i++) if (f[i] + g[i] == ans) {
        tt++; if (dep[i] > dep[top]) top = i;
    }
    for (int e = adj[top]; e; e = nxt[e]) {
        if (f[go[e]] + val[e] == f[top]) {
            cnt1++;
            if (cnt1 == 1) u1 = go[e];
            else if (cnt1 == 2) u2 = go[e];
        }
        if (f[go[e]] + val[e] == g[top]) cnt2++, v = go[e];
    }
    if (cnt1 >= 3) Ans = 0;
    else if (cnt1 == 2) Ans = 2, dfs2(u1, top), dfs2(u2, top);
    else {
        if (cnt2 >= 2 || cnt2 == 0) Ans = 1, dfs2(u1, top);
        else {
            Ans = 1; dfs2(u1, top);
            if (tt == 1) Ans++, dfs2(v, top);
        }
    }
    printf("%d\n", Ans);
}
int main() {
    int i, x, y, z; n = read();
    for (i = 1; i < n; i++) x = read(), y = read(), z = read(),
        add_edge(x, y, z); solve();
    return 0;
}