树的中心 —— 树型DP

题面

题目

思路

因为要求某个点到其他结点的最远距离最近；

那么我们只需要处理出每个结点的最远距离从中取一个min即可；

---

取某个结点的最远距离

• 可以从该结点的上方转移过来

• 也可以从该结点的下方转移过来；

---

首先讨论从下方转移过来(由子节点转移到父节点)

因为题目要求最远的，因此我们只需要维护一个最远的距离即可；

如下所示

//u是当前结点,to是子节点
max(dfs_d(to,u) + e[i].val)

接着考虑从上方转移过来(由父节点转移到子节点)

比如说我们要由结点$1$更新结点$2$

那么对于结点$2$来说，它往上更新肯定要走$2\to 1$这条边；

即$2\to 1+X$

这个$X$的取值有两种情况，一种是上面转移过来，一种是从同级的转移过来($2$的兄弟，$1$的儿子)；

如果从兄弟转移过来，我们不能让兄弟等于二号结点，因此我们还要维护一个往下的次大值；

如下所示

//判断往下的最大距离是否从to过来
 if(p1[u] == to){
      //只能从次大值更新
      u1[to] = max(u1[u],d2[u]) + e[i].val;
  }
  else u1[to] = max(u1[u],d1[u]) + e[i].val;

完整Code

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N = 1e4 + 10;

int head[N],cnt,root;

//当前点往下走的最大值，次大值；当前点往上走的最大值
int d1[N],d2[N],u1[N];

int p1[N];//当前点往下走的最大值是从哪个点更新过来的

bool is_leaf[N];//是否为叶子节点，如果是只能从up数组更新(只能往上走)

struct Edge{
    int val,next,to;
}e[N<<1];

void add(int u,int v,int w){
    ++cnt;
    e[cnt].to = v;
    e[cnt].val = w;
    e[cnt].next = head[u];
    head[u] = cnt;
}

//往下的最大值(由子节点更新父节点)
int dfs_d(int u,int fa){
    for(int i=head[u];i;i=e[i].next){
        int to = e[i].to;
        if(fa == to) continue;
        int d = dfs_d(to,u) + e[i].val;
        if(d > d1[u]){
            d2[u] = d1[u],d1[u] = d;
            p1[u] = to;
        }
        else if(d > d2[u]) d2[u] = d;
    }
    if(d1[u] == 0){
        //说明是叶子节点，因为边权至少是1
        is_leaf[u] = 1;
    }
    return d1[u];
}

//往上的最大值(由父节点更新子节点)
void dfs_u(int u,int fa){
    for(int i=head[u];i;i=e[i].next){
        int to = e[i].to;
        if(to == fa) continue;
        //判断往下的最大距离是否从to过来
        if(p1[u] == to){
            //只能从次大值更新
            u1[to] = max(u1[u],d2[u]) + e[i].val;
        }
        else u1[to] = max(u1[u],d1[u]) + e[i].val;
        dfs_u(to,u);
    }
}
void solve(){
    int n;
    cin >> n;
    for(int i=1,u,v,w;i<n;++i){
        cin >> u >> v >> w;
        add(u,v,w),add(v,u,w);
    }
    root = 1;
    dfs_d(root,-1);
    dfs_u(root,-1);
    //root不能往上
    int ans = d1[root];
    for(int i=2;i<=n;++i){
        //叶子不能往下
        if(is_leaf[i]) ans = min(ans,u1[i]);
        else ans = min(ans,max(u1[i],d1[i]));
    }
    cout << ans << '\n';
}

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