一所学校不久前买了第一台电脑(所以这台电脑的ID是1)。近年来,学校购买了N-1新电脑。每台新电脑都连接到一台先前安装的电脑上。学校的管理人员担心网络运行缓慢,希望知道第i台计算机需要发送信号的最大距离si(即到最远计算机的电缆长度)。您需要提供此信息。
提示:示例输入与此图对应。从图中,你可以看到计算机4离1最远,所以s1=3。计算机4和5是距离2最远的,所以s2=2。计算机5是离3最远的,所以s3=3。我们也得到了s4=4,s5=4。输入输入文件包含多组测试样例。在每组样例中,第一行中都有自然数n(n<=10000),然后是(n-1)行,其中包含对计算机的描述。第i行包含两个自然数-第i计算机所连接的计算机和用于连接的电缆长度。电缆总长度不超过10^9。输入行中的数字用空格分隔。输出对于每组样例,输出n行。第i行第i台计算机的到其他计算机的最大长度Si(1<=i<=n)。样例输入
5 1 1 2 1 3 1 1 1
样例输出
3 2 3 4 4
提示示例输入与此图对应。从图中,你可以看到计算机4离1最远,所以s1=3。计算机4和5是距离2最远的,所以s2=2。计算机5是离3最远的,所以s3=3。我们也得到了s4=4,s5=4。
思路:就是用3次搜索,第一次随便找一个点,第一次BFS,寻找第一个距离该点最远的点记为端点point1,第二次搜索以端点point1寻找端点point2,那么point2和point1就是距离根节点最远的点。那么每个点距离最远的点要么是point1要么是point2,所以在搜索的时候
用数组记录仪一下每个点到point1和到point2的距离就可以啦。最后输出,,谁大输出谁。。
#include<iostream> #include<queue> #include<vector> #include<cstring> using namespace std; typedef long long ll; struct stu{ int y,s; }; int mark[10000+10]; int step[10000+10]; int step2[10000+10]; vector<stu>ve[10000+10]; int ans=0; int point; void bfs(int x){ queue<int >que; que.push(x); mark[x]=1; step[x]=0; while(que.size()){ int xx=que.front(); que.pop(); if(step[xx]>ans){ ans=step[xx]; point =xx;//寻找两个树点端点; } for(int i=0;i<ve[xx].size();i++){ if(mark[ve[xx][i].y]==0){ mark[ve[xx][i].y]=1; step[ve[xx][i].y]=step[xx]+ve[xx][i].s; que.push(ve[xx][i].y); } } } }
int main() { int n; while(scanf("%d",&n)!=EOF){ for(int i=2;i<=n;i++){ int a,b; scanf("%d %d",&a,&b); ve[i].push_back({a,b}); ve[a].push_back({i,b}); } ans=0;//初始化,再每次搜索前都要初始化,。。。。 memset(mark,0,sizeof(mark)); bfs(1); int point1=point;//端点1 memset(mark,0,sizeof(mark)); ans=0;//没有初始化哇了好几次 bfs(point1); for(int i=1;i<=n;i++){ step2[i]=step[i]; } int point2=point ;//端点2; memset(mark,0,sizeof(mark)); ans=0; bfs(point2); for(int i=1;i<=n;i++){ printf("%lld ",max(step[i],step2[i])); } for(int i=1;i<=n;i++){ ve[i].clear(); } } return 0; }