问题描述
给定一棵树,树中每条边都有一个权值,树中两点之间的距离定义为连接两点的路径边权之和。树中最远的两个节点(两个节点肯定都是叶子节点)之间的距离被称为树的直径,连接这两点的路径被称为树的最长链。后者通常也可称为直径。
算法:
不妨设1号节点为根,“N个节点,N-1条边的无向图”就可以看作“有根树”。设d[x]表示从节点x出发走向以x为根的子树,能够到达最远节点的距离。设x的子节点为y1,y2,……,yt,edge(x, y)表示边权,显然有:
接下来,我们可以考虑对每个节点x求出“经过节点x的最长链的长度”ans[x],整棵树的直径就是:
那么,如何求ans[x]呢?我们用链式前向星存图,依次遍历以x为起点的所有连边,用已经更新过的d[x]与没有枚举到的x的连边之和来更新ans[x]:
注意ans[i]存的是经过i节点的最长链,最后还要便利一遍ans才能求到真正的直径
而我们的ans数组其实可以用一个整型数ans代替,就不需要后面再更新一遍了
关键代码:
void dp(int x){
vis[x]=1;
for(int i=head[x];i;i=edge[i].next){
int y=edge[i].to;
if(vis[y]) continue;
dp(y)