求树的直径的两种方式

定义：树中距离最远的两个点的距离

 

1.双bfs求直径，任意找一个点O，找到离O距离最远的点P，再求出离P最远的点的距离就是树的直径

int bfs(int u)
{
    mesmet(d, -1, sizeof d);
    int hh = 0, tt = 0;
    d[u] = 0, q[0] = u;
    
    while(hh <= tt)
    {
        int t = q[hh ++];
        
        for(int i = h[u]; ~i; i = ne[i])
        {
            int j = e[i];
            if(d[j] == -1)
            {
                d[j] = d[t] + 1;
                q[++ tt] = j;
            }
        }
    }
    
    for(int i = 1; i <= n; i ++ ) if(d[i] > d[u]) u = i;
    return u;
}

 

2.dfs求直径

//任意找一点作为dfs起点即可
int dfs(int u, int fa)
{
    int d1 = 0, d2 = 0;
    for(int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if(j == fa) continue;
        int d = dfs(j, u) + w[i];
        if(d > d1) d2 = d1, d1 = d;
        else if(d > d2) d2 = d;
    }
    res = max(res, d1 + d2); // 最终res是直径的长度
    return d1;
}

 

 

3.dp求直径