氪金带东
题意:
实验室里原先有一台电脑(编号为1),最近氪金带师咕咕东又为实验室购置了N-1台电脑,编号为2到N。每台电脑都用网线连接到一台先前安装的电脑上。但是咕咕东担心网速太慢,他希望知道第i台电脑到其他电脑的最大网线长度,但是可怜的咕咕东在不久前刚刚遭受了宇宙射线的降智打击,请你帮帮他。
![](https://s1.ax1x.com/2020/03/25/8XylFS.png)
提示:
![](https://s1.ax1x.com/2020/03/25/8XylFS.png)
提示:
样例输入对应这个图,从这个图中你可以看出,距离1号电脑最远的电脑是4号电脑,他们之间的距离是3。 4号电脑与5号电脑都是距离2号电脑最远的点,故其答案是2。5号电脑距离3号电脑最远,故对于3号电脑来说它的答案是3。同样的我们可以计算出4号电脑和5号电脑的答案是4.
Input
输入文件包含多组测试数据。对于每组测试数据,第一行一个整数N (N<=10000),接下来有N-1行,每一行两个数,对于第i行的两个数,它们表示与i号电脑连接的电脑编号以及它们之间网线的长度。网线的总长度不会超过10^9,每个数之间用一个空格隔开。
Output
对于每组测试数据输出N行,第i行表示i号电脑的答案 (1<=i<=N).
Sample Input
5 1 1 2 1 3 1 1 1
Sample Output
3 2 3 4 4
我的题解:
先求树的直径:
从任一点出发dfs找到最远的叶子节点,在从这个叶子节点出发dfs找到距离最远的叶子节点。这两个叶子节点之间的距离即为直径。
求树中某节点的最长路径:
只需要分别从直径叶子节点出发遍历然后找到距离该节点的距离,取较大值即可!
1 #include<iostream> 2 #include<stdio.h> 3 #include<algorithm> 4 #include<string.h> 5 using namespace std; 6 7 const int maxn=100005; 8 int n,maxL,maxlen,f1,f2; 9 int L1[maxn],L2[maxn]; 10 int u,w; 11 typedef struct node{ 12 int u,v,w,nxt; 13 }Edge; 14 Edge Edges[maxn]; 15 int head[maxn],tot,vis[maxn]; 16 17 void addEdge(int u,int v,int w){ 18 Edges[tot].u=u; 19 Edges[tot].v=v; 20 Edges[tot].w=w; 21 Edges[tot].nxt=head[u]; 22 head[u]=tot; 23 tot++; 24 } 25 26 void dfs(int u,int *a,int L){ 27 a[u]=L; 28 if(maxL<L) { 29 maxL=L;maxlen=u; 30 } 31 for(int i=head[u];i!=-1;i=Edges[i].nxt){ 32 if(!vis[Edges[i].v]){ 33 vis[Edges[i].v]=true; 34 dfs(Edges[i].v, a, L + Edges[i].w); 35 } 36 } 37 } 38 int main(){ 39 while(cin>>n){ 40 41 tot=0; 42 for(int i=0;i<n+2;++i) head[i]=-1; 43 44 for(int i=2;i<=n;++i){ 45 cin>>u>>w; 46 addEdge(i,u,w); 47 addEdge(u,i,w); //加入双向边 48 } 49 50 memset(vis,0,sizeof(vis)); 51 maxL=-1;vis[1]=1; 52 53 dfs(1,L1,0); 54 f1=maxlen; 55 memset(vis,0,sizeof(vis)); 56 maxL=-1; 57 vis[f1]=1; 58 59 dfs(f1,L1,0); 60 f2=maxlen; 61 memset(vis,0,sizeof(vis)); 62 vis[f2]=1; 63 dfs(f2,L2,0); 64 65 for(int i=1;i<=n;i++){ 66 cout<<max(L1[i],L2[i])<<endl; 67 //较大者即为输出 68 } 69 } 70 return 0; 71 }
流转星云