浅谈关于树形dp求树的直径问题

在一个有n个节点，n-1条无向边的无向图中，求图中最远两个节点的距离，那么将这个图看做一棵无根树，要求的即是树的直径。

求树的直径主要有两种方法：树形dp和两次bfs/dfs，因为我太菜了不会写后者这里只介绍树形dp

• 树形dp求树的直径
  我们不妨设1号点为根节点，那么这就可以看做一棵有根树。
  设D[x]表示从节点x出发，往以x为根的子树走，能够到达的最远距离。设x的子节点分别为\(y_1,y_2,y_3,...,y_t\)，\(edge(x,y)\)表示从x到y的边权，则可以得到状态转移方程：
  \(D[x]=max{D[y_i]+edge(x,y_i)}\)
  接下来，我们考虑对于每个节点x求出经过x的最长链的长度F[x],整棵树的直径就是max{F[x]}(1<=x<=n)。

  现在我们考虑如何求F[x]。
  对于任意两个节点yi和yj，经过节点x的最长链的长度可以通过四个部分来构成：
  • D[yi]
  • D[yj]
  • 从x到yi的距离
  • 从x到yj的距离

  不妨设j<i，则有：

\(F[x]= max{D[y_i]+D[y_j]+edge(x,y_i)+edge(x,y_j)}\)

对应代码如下：

void dp(int x){
    v[x]=1; 
    for(register int i=head[x];i;i=nxt[i]){
        int y=ver[i];
        if(v[y])continue;   
        dp(y); 
        ans=max(ans,d[x]+d[y]+edge[i]);
        d[x]=max(d[x],d[y]+edge[i]);
    }
}

代码解释可以看图：

参考资料：李煜东《算法竞赛进阶指南》