Week 6 氪金带东

题目描述

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

tips： 样例输入对应这个图，从这个图中你可以看出，距离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).

样例输入

5
1 1
2 1
3 1
1 1

样例输出

3
2
3
4
4

思路

假设树的直径是叶子节点v1、v2之间的距离，本题要找某个节点x所能到达的最长路径，那么这个节点x 的最长路径要么是到v1的路径、要么是到v2的路径
使用前向星存储树中的边，使用DFS进行遍历，首先我们要确定树的直径
第一次遍历，从任意一个节点出发，找到距离该节点最远的节点v1
第二次遍历，从v1结点出发，找到距离v1最远的节点，即v2，并用path1记录每个节点到v1的距离，此时已求出树的直径
第三次遍历，从v2出发，使用path2记录每个节点到v2的最大距离
只需要取节点到v1、v2中大的距离即可

在寻找v1、v2过程中，要使用一个变量记录当前距离的最大值，并实时更新

总结

在开edge数组时，忽略无向图应该是两个点之间存两条边，只开了一条边的空间，导致wa

每进行一次dfs时要注意将vis[]初始化

代码

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

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

int path[10010],path1[10010],path2[10010];
int head[10010];
bool vis[10010];
int size=0;
int n1,n2,v1,v2;


void initial(int n)
{
	size=0;
	n1=n2=0;
	for(int i=1;i<=n;i++)
	{
		head[i]=-1;
		vis[i]=false;
		path[i]=0;
		path1[i]=0;
		path2[i]=0;
	}
}

void addEdge(int u,int v,int w)
{
	edge[size].u=u;
	edge[size].v=v;
	edge[size].w=w;
	edge[size].next=head[u];
	head[u]=size;
	size++;
}

void dfs1(int u)//第一次遍历 
{
	vis[u]=true;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int temp=edge[i].v;
		if(!vis[temp])
		{
			vis[temp]=true;
			path[temp]=path[u]+edge[i].w;
			if(path[temp]>n1)
			{
				n1=path[temp];
				v1=temp;
			}
			dfs1(temp);
		}
	}
}

void dfs2(int u)
{
	vis[u]=true;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int temp=edge[i].v;
		if(!vis[temp])
		{
			vis[temp]=true;
			path1[temp]=path1[u]+edge[i].w;
			if(path1[temp]>n2)
			{
				n2=path1[temp];
				v2=temp;
			}
			dfs2(temp);
		 } 
	}
}

void dfs3(int u)
{
	vis[u]=true;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int temp=edge[i].v;
		if(!vis[temp])
		{
			vis[temp]=true;
			path2[temp]=path2[u]+edge[i].w;
			dfs3(temp);
		}
	}
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
//		n1=n2=0;
//		memset(head,-1,sizeof(head));
//		memset(vis,false,sizeof(vis));
//		memset(path,0,sizeof(path));
//		memset(path1,0,sizeof(path1));
//		memset(path2,0,sizeof(path2));
		initial(n);
		
		for(int i=2;i<=n;i++)
		{
			int u,w;
			cin>>u>>w;
			addEdge(u,i,w);
			addEdge(i,u,w);
		}
		dfs1(0);
		memset(vis,false,sizeof(vis));
		dfs2(v1);
		memset(vis,false,sizeof(vis));
		dfs3(v2);
		for(int i=1;i<=n;i++)
		{
			if(path1[i]>path2[i])
				cout<<path1[i]<<endl;
			else 
				cout<<path2[i]<<endl;
		}
	} 
 }