网络铺设（树的直径求解例题）

网络铺设（树的直径）

问题描述

你现在是城市的主人
现在有 n 个村庄，已经修建了 n−1 条道路，使得各个村庄作为节点，路作为边，构成一棵树。
假设第 a 个村庄到第 b 个村庄有路相连，则从 a 走到 b 或者从 b 走到 a 需要走 1m 。

你需要输出 n 个数，第 i 个数代表从第 i 个村庄出发，距离他最远的村庄的距离

输入格式

第一行一个整数 n , 0≤n≤105 接下来有 n−1 行，每行 a,b 代表第 a 个村庄，到第 b 个村庄有一条路。 1≤a,b≤105

保证输入结构是一棵树

输出格式

输出一行 n 个数，表示答案

样例输入

5
5 1
1 2
2 3
3 4

样例输出

3 2 3 4 4

算法

从一个随机点开始，找到距离这个随机点最远的点p1，再从p1开始，找到距离p1最远的点p2，则p1p2就是直径

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int vis[1000000] = {0};
ll dis[1000000] = {0};
ll d1[1000000] = {0};
ll d2[1000000] = {0};

int main()
{
     ll n;
     cin >> n;
     vector<vector<ll>> tu(n + 1);
     for (ll i = 1; i <= n - 1; i++)
     {
          ll a, b;
          cin >> a >> b;
          tu[a].push_back(b);
          tu[b].push_back(a);
     }

     //设直径是p1-p2
     //找直径上的p1
     ll p1, p2;   //直径的左右端点
     queue<ll> q; //使用BFS，找到的到每一个点的距离一定是最小
     q.push(1);
     vis[1] = 1;
     while (!q.empty())
     {
          ll root = q.front();
          q.pop();
          for (ll i = 0; i < tu[root].size(); i++)
          {
               if (vis[tu[root][i]] == 0)
               {
                    vis[tu[root][i]] = 1;
                    q.push(tu[root][i]);
                    dis[tu[root][i]] = dis[root] + 1;
               }
          }
     }

     //找p1
     ll max_length = 0; // p1是与用来找原点的那个点距离最远的点
     for (ll i = 1; i <= n; i++)
     {
          vis[i] = 0;
          if (max_length < dis[i])
          {
               max_length = dis[i];
               p1 = i;
          }
     }

     //找p2,顺便记录p1到所有点的值
     q.push(p1);
     vis[p1] = 1;
     while (!q.empty())
     {
          ll root = q.front();
          q.pop();
          for (ll i = 0; i < tu[root].size(); i++)
          {
               if (vis[tu[root][i]] == 0)
               {
                    vis[tu[root][i]] = 1;
                    q.push(tu[root][i]);
                    d1[tu[root][i]] = d1[root] + 1;
               }
          }
     }
     max_length = 0;
     for (ll i = 1; i <= n; i++)
     {
          vis[i] = 0;
          if (max_length < d1[i])
          {
               max_length = d1[i];
               p2 = i;
          }
     }

     //从p2往回找
     q.push(p2);
     vis[p2] = 1;
     while (!q.empty())
     {
          ll root = q.front();
          q.pop();
          for (ll i = 0; i < tu[root].size(); i++)
          {
               if (vis[tu[root][i]] == 0)
               {
                    vis[tu[root][i]] = 1;
                    q.push(tu[root][i]);
                    d2[tu[root][i]] = d2[root] + 1;
               }
          }
     }

     for (ll i = 1; i <= n; i++)
     {
          cout << max(d1[i], d2[i]) << ' ';
     }
     return 0;
}