week6_A 氪金带东（树的直径）

题目概述

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

输入输出

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

思路分析

本题求每台电脑的最大网线长度，借助树的直径解决。

树的直径是指树中任意两点之间距离的最大值。假设直径的端点分别为v1和v2，那么对于每台电脑来说，最大网线长度就是该电脑到v1或v2的网线长度。因此，需要根据输入构建图，然后进行三次遍历：第一次遍历找到直径所在路线（从任意一个点开始遍历，这里直接选择编号为1的点），第二次遍历从v1到该台电脑，第三次遍历从v2到该台电脑。比较第二次和第三次遍历的长度，输出两者中最大值。

反思

printf输出不用“&”取地址，不取地址！不取地址！重要事情讲三遍。写了太多cout，printf都不会用了…

代码

#include<iostream>
#include<stdio.h>
using namespace std;

int n,tot,v1,v2,ans1,ans2,ans3;
//dis数组记录距离
int dis1[10005],dis2[10005],dis3[10005],head[10005];
bool vis1[10005],vis2[10005],vis3[10005];//标记是否到达

struct Edge{
	int u,v,w,nxt;
}edge[20010];

void init()
{//初始化 
 tot=0;
 ans1=0;
 ans2=0;
 ans3=0;  
 for(int i=1;i<=n;i++)
 {
  head[i]=-1;
  dis1[i]=0;
  dis2[i]=0;
  dis3[i]=0;
  vis1[i]=0;
  vis2[i]=0;
  vis3[i]=0;
 }
 for(int i=0;i<n;i++)
  edge[i].nxt=-1;
} 

void addEdge(int u,int v,int w)
{
 edge[tot].u=u;
 edge[tot].v=v;
 edge[tot].w=w;
 edge[tot].nxt=head[u];//总在头部插入 
 head[u]=tot;
 tot++;
}

void dfs1(int u)
{
 vis1[u]=true;
 for(int i=head[u];i!=-1;i=edge[i].nxt)
 {
  if(!vis1[edge[i].v])
  {
   vis1[edge[i].v]=true;
   dis1[edge[i].v]=dis1[u]+edge[i].w;
   if(dis1[edge[i].v]>ans1) 
   {
    v1=edge[i].v;
    ans1=dis1[edge[i].v];
   }
   dfs1(edge[i].v);
  }
 }
}

void dfs2(int u)
{
 vis2[u]=true;
 for(int i=head[u];i!=-1;i=edge[i].nxt)
 {
  if(!vis2[edge[i].v])
  {
   vis2[edge[i].v]=true;
   dis2[edge[i].v]=dis2[u]+edge[i].w;
   if(dis2[edge[i].v]>ans2) 
   {
    v2=edge[i].v;
    ans2=dis2[edge[i].v];
   }
   dfs2(edge[i].v);
  }
 }
}

void dfs3(int u)
{
 vis3[u]=true;
 for(int i=head[u];i!=-1;i=edge[i].nxt)
 {
  if(!vis3[edge[i].v])
  {
   vis3[edge[i].v]=true;
   dis3[edge[i].v]=dis3[u]+edge[i].w;
   if(dis3[edge[i].v]>ans3) 
   {
    //v3=edge[i].v;
    ans3=dis3[edge[i].v];
   }
   dfs3(edge[i].v);
  }
 }
}

int main()
{
 
 while(cin>>n)
 {
  init();
  for(int i=2;i<=n;i++)
  {
   int v,w;
   cin>>v>>w;
   addEdge(i,v,w);
   addEdge(v,i,w);
  }  
  dfs1(1);
  dfs2(v1);
  dfs3(v2);
  for(int i=1;i<=n;i++)
  {
   //if(dis2[i]>dis3[i]) printf("%d\n",dis2[i]);
   //else printf("%d\n",dis3[i]);//printf不取地址，又忘了！！ 
   int ans=max(dis2[i],dis3[i]);
   cout<<ans<<endl;
  }  
 }
 return 0;
}