【Jeoy‘ s daily 】AcWing 1073. 树的中心 (树形DP)

首先给一下题面

给定一棵树，树中包含 n 个结点（编号1~n）和 n−1 条无向边，每条边都有一个权值。
请你在树中找到一个点，使得该点到树中其他结点的最远距离最近。

输入格式
第一行包含整数 n。
接下来 n−1 行，每行包含三个整数 ai,bi,ci，表示点 ai 和 bi 之间存在一条权值为 ci 的边。

输出格式
输出一个整数，表示所求点到树中其他结点的最远距离。

数据范围

1≤n≤10000,
1≤ai,bi≤n,
1≤ci≤105

输入样例：

5 
2 1 1 
3 2 1 
4 3 1 
5 1 1

输出样例：

2

大概思路
这道题目大致做法就是

1. 枚举每一个点，找到向下的最大路径
2. 再次枚举每一个点，跟新向上的最大路径 （注意 ： 此时因为向上节点的最大值可能是需要经过本节点，所以需要存一个次大值备用）
3. 再次再次再次枚举每个点，找到最小的最大值

代码部分

加边函数

void add ( int a, int b, int c ) {
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}

向下搜索函数

int dfs_d ( int p, int father ) {
    d1[p] = d2[p] = -INF;
    int dist = 0;

    for ( int i = h[p]; ~i; i = ne[i] ) {
        int t = e[i];
        if ( t == father ) continue;

        int d = dfs_d ( t, p ) + w[i];
        if ( d >= d1[p] ) {
            d2[p] = d1[p], p2[p] = p1[p];
            d1[p] = d, p1[p] = t;
        }
        else if ( d > d2[p] ) {
            d2[p] = d, p2[p] = t;
        }

        dist = max ( dist, d );
    }
    return dist;
}

向上搜索

int dfs_u ( int p, int father ) {
    for ( int i = h[p]; ~i; i = ne[i] ) {
        int t = e[i];
        if ( t == father ) continue;
        if ( p1[p] == t ) up[t] = max ( up[p], d2[p]  ) + w[i];
        else up[t] = max ( up[p], d1[p]  ) + w[i];
        dfs_u( t, p );
    }
}

找最小值

int res = INF;
    for ( int i = 1; i <= n; i ++ ) {
        res = min ( res, max ( up[i], d1[i] ) );
       // cout << max ( up[i], d1[i] ) << endl;
    }

    cout << res << endl;

完整代码

#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f;
const int N = 10010, M = N * 2;
int idx, h[N], e[M], ne[M], w[M], n;
int up[N]; //储存向上找的最大
int d1[N], d2[N], p1[N], p2[N];

void add ( int a, int b, int c ) {
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}

int dfs_u ( int p, int father ) {
    for ( int i = h[p]; ~i; i = ne[i] ) {
        int t = e[i];
        if ( t == father ) continue;
        if ( p1[p] == t ) up[t] = max ( up[p], d2[p]  ) + w[i];
        else up[t] = max ( up[p], d1[p]  ) + w[i];
        dfs_u( t, p );
    }
}

int dfs_d ( int p, int father ) {
    d1[p] = d2[p] = -INF;
    int dist = 0;

    for ( int i = h[p]; ~i; i = ne[i] ) {
        int t = e[i];
        if ( t == father ) continue;

        int d = dfs_d ( t, p ) + w[i];
        if ( d >= d1[p] ) {
            d2[p] = d1[p], p2[p] = p1[p];
            d1[p] = d, p1[p] = t;
        }
        else if ( d > d2[p] ) {
            d2[p] = d, p2[p] = t;
        }

        dist = max ( dist, d );
    }
    return dist;
}

int main () {
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);

    memset ( h, -1, sizeof h );
    cin >> n;
    for ( int i = 1; i <= n - 1; i ++ ) {
        int a, b, c;
        cin >> a >> b >> c;
        add ( a, b, c ), add( b, a, c );
    }
    dfs_d ( 1, -1 );
    dfs_u ( 1, -1 );

    int res = INF;
    for ( int i = 1; i <= n; i ++ ) {
        res = min ( res, max ( up[i], d1[i] ) );
       // cout << max ( up[i], d1[i] ) << endl;
    }

    cout << res << endl;

}