week6作业A 氪金带东 bfs 树直径

题目:
实验室里原先有一台电脑(编号为1),最近氪金带师咕咕东又为实验室购置了N-1台电脑,编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。但是咕咕东担心网速太慢,他希望知道第i台电脑到其他电脑的最大网线长度,但是可怜的咕咕东在不久前刚刚遭受了宇宙射线的降智打击,请你帮帮他。

提示: 样例输入对应这个图,从这个图中你可以看出,距离1号电脑最远的电脑是4号电脑,他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点,故其答案是2。5号电脑距离3号电脑最远,故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.

输入:
输入文件包含多组测试数据。对于每组测试数据,第一行一个整数N (N<=10000),接下来有N-1行,每一行两个数,对于第i行的两个数,它们表示与i号电脑连接的电脑编号以及它们之间网线的长度。网线的总长度不会超过10^9,每个数之间用一个空格隔开。
在这里插入图片描述
输出:
对于每组测试数据输出N行,第i行表示i号电脑的答案 (1<=i<=N).
样例输入:
5
1 1
2 1
3 1
1 1
样例输出:
3
2
3
4
4

在解决这道题前,要先了解如何求树的直径。求树的直径需要用两次bfs/dfs,第一次从任一点出发,找到离该点最远的点,然后再从找到的最远点开始,进行一次bfs/dfs,找到离该点最远的点,这两个点就是树的直径的两个端点。
但是本题不是找直径,是找某个节点x所能到达的最长路径,那么最长路径的第二个节点就是直径的两个端点之一,因为不能确定是到哪个端点,所以还需要一次bfs/dfs。
本题的数据结构用邻接链表。

以下是完整代码:

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
struct edge{
 int v,w;
 edge(int vx,int wx)
 {
  v=vx;
  w=wx;
 }
 edge()
 {
 }
};
int reach[11000];//bfs过程中用来标记是否已经访问过 
long long ans1[11000],ans2[11000],d[11000],ans[11000];//前三个用于bfs过程中存储距离,ans用来存储最后答案 
int N;
vector<edge> a[11000];
void bfs(int x)
{
 for(int i=0;i<=N;i++)
 {
  reach[i]=0;
  d[i]=0;
 }
 queue<int> q;
 q.push(x);
 int temp;
 reach[x]=1;
 while(!q.empty())
 {
  temp=q.front();
  q.pop();
  for(int i=0;i<a[temp].size();i++)
  {
   edge next=a[temp][i];
   if(reach[next.v]==0)
   {
    d[next.v]=d[temp]+next.w; 
    q.push(next.v); 
    reach[next.v]=1;
   }
  }
 } 
}
void bfs1(int x)//求出第二个端点,同时求出所有点到第一个端点的距离 
{
 for(int i=0;i<=N;i++)
 {
  reach[i]=0;
  ans1[i]=0;
 }
 queue<int> q;
 q.push(x);
 int temp;
 reach[x]=1;
 while(!q.empty())
 {
  temp=q.front();
  q.pop();
  for(int i=0;i<a[temp].size();i++)
  {
   edge next=a[temp][i];
   if(reach[next.v]==0)
   {
    ans1[next.v]=ans1[temp]+next.w; 
    q.push(next.v); 
    reach[next.v]=1;
   }
  }
 } 
}
void bfs2(int x)//同时求出所有点到第二个端点的距离 
{
 for(int i=0;i<=N;i++)
 {
  reach[i]=0;
  ans2[i]=0;
 }
 queue<int> q;
 q.push(x);
 int temp;
 reach[x]=1;
 while(!q.empty())
 {
  temp=q.front();
  q.pop();
  for(int i=0;i<a[temp].size();i++)
  {
   edge next=a[temp][i];
   if(reach[next.v]==0)
   {
    ans2[next.v]=ans2[temp]+next.w; 
    q.push(next.v); 
    reach[next.v]=1;
   }
  }
 }  
}
void solve()
{
 bfs(1);
 int maxv=0;
 int k=0;
 for(int i=1;i<=N;i++)//求出直径的一个端点 
 {
  if(maxv<d[i])
  {
   maxv=d[i];
   k=i;
  }
 }
 int max1=k;
 bfs1(max1);
 maxv=0;
 k=0;
 for(int i=1;i<=N;i++)//求出直径的第二个端点 
 {
  if(maxv<ans1[i])
  {
   maxv=ans1[i];
   k=i;
  }
 }
 int max2=k;
 bfs2(max2);
 for(int i=1;i<=N;i++)//把ans1和ans2比较,取较大值放入ans数组 
 {
  if(ans1[i]<ans2[i])
  {
   ans[i]=ans2[i];
  }
  else
  {
   ans[i]=ans1[i];
  }
 }
 for(int i=1;i<=N;i++)
 {
  cout<<ans[i]<<'\n';
 }
}
int main()
{
 while(cin>>N)
 {
  for(int i=2;i<=N;i++)
  {
   int v,w;
   cin>>v>>w;
   a[i].push_back(edge(v,w));
   a[v].push_back(edge(i,w));  
  }
  solve();
  for(int i=1;i<=N;i++)//把之前的数据清空 
  {
   a[i].clear();
  }
 }
}
发布了24 篇原创文章 · 获赞 5 · 访问量 268

猜你喜欢

转载自blog.csdn.net/qq_45639151/article/details/105142583
今日推荐