网络铺设(树的直径)
问题描述
你现在是城市的主人
现在有 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;
}