氪金带东 Week6作业A题

题目：

实验室里原先有一台电脑(编号为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).

样例：
Input:

5
1 1
2 1
3 1
1 1

Output:

3
2
3
4
4

思路：

这道题不是找最长路径，是找某个节点x所能到达的最长路径。
求树的直径的方法：
假设树的最长路的两个叶子节点为v1,v2，从任意一点u出发走到的最 远的点一定是（v1,v2）中的一点，然后再从v1或者v2出发走到的最 远点一定是v2或者v1。由此经过两次遍历就能找到最长路径。

所以节点x的最长路径要么是到v1的路径，要么就是到v2的路径。由于不知道往哪边走才是最长的路径，所以需要分别从v1,v2点再次遍历共需要三次遍历。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int n,v1,v2,tot,ans1,ans2,ans3,head[10050];
int fl1[10050],fl2[10050],fl3[10050];
bool flag1[10050],flag2[10050],flag3[10050];

struct node
{
	int u;
	int v;
	int w;
	int next;
}e[20020];

void addn(int u,int v,int w)
{
	e[++tot].v=v;
	e[tot].w=w;
	e[tot].next=head[u];
	head[u]=tot;
}

void dfs1(int u)
{
	flag1[u]=1;
	for(int i=head[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(!flag1[v])
		{
			flag1[v]=1;
			fl1[v]=fl1[u]+e[i].w;
			if(fl1[v]>ans1)
			{
				ans1=fl1[v];
				v1=v;
			}
			dfs1(v);
		}
	}
}

void dfs2(int u)
{
	flag2[u]=1;
	for(int i=head[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(!flag2[v])
		{
			flag2[v]=1;
			fl2[v]=fl2[u]+e[i].w;
			if(fl2[v]>ans2)
			{
				ans2=fl2[v];
				v2=v;
			}
			dfs2(v);
		}
	}
}

void dfs3(int u)
{
	flag3[u]=1;
	for(int i=head[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(!flag3[v])
		{
			flag3[v]=1;
			fl3[v]=fl3[u]+e[i].w;
			if(fl3[v]>ans3)
			    ans3=fl2[v];
			dfs3(v);
		}
	}
}

int main()
{
	int v,w;
	while(scanf("%d",&n)!=EOF)
	{
		memset(flag1,0,sizeof(flag1));
		memset(flag2,0,sizeof(flag2));
		memset(flag3,0,sizeof(flag3));
		memset(fl1,0,sizeof(fl1));
		memset(fl2,0,sizeof(fl2));
		memset(fl3,0,sizeof(fl3));
		memset(head,0,sizeof(head));
		ans1=0,ans2=0,ans3=0,tot=0;
		
		for(int i=2;i<=n;i++)
		{
			//cin>>v>>w;
			scanf("%d%d",&v,&w);
			addn(i,v,w);
			addn(v,i,w);
		}
		dfs1(1);
		dfs2(v1);
		dfs3(v2);
		for(int i=1;i<=n;i++)
		    //cout<<max(fl2[i],fl3[i])<<endl;
		    printf("%d\n",max(fl2[i],fl3[i]));
	}
	return 0;
}