SDU_week6_A - 氪金带东（三次dfs求所有点最远值）

题目描述

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

提示: 样例输入对应这个图，从这个图中你可以看出，距离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

题目分析

本题要求第i台电脑到其他电脑的最大网线长度，naive做法是在每台电脑跑一个dfs（注意求最远用dfs，求最近用bfs效果相对好），一共跑结点数次，慢，不可取。
考虑这样一个问题：距离任意节点的最远点是否为树的直径（树宽）两点中的一点？答案是肯定的。所以通过三次dfs即可解决问题。

1. 第一次从任意点开始dfs，找到其中一个最远点v1
2. 第二次从v1开始dfs，找v2的过程中记录所有点到v1的距离
3. 第三次从v2开始dfs，记录所有点到v2的距离

那么最终答案就是第二次和第三次的距离取最大值。

注意

每次dfs后要重置标记数组以及当前的最大值MAX

代码

#define _CRT_SECURE_NO_WARNINGS
#define _ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

#include <bits/stdc++.h>
//#include <iostream>
//#include<algorithm>

using namespace std;

int N;//节点数
int tot;//边数计数变量，本题为树结构，tot=N-1

//链式前向星存图
int head[10009];//头节点
struct edge
{//边集
	int u, v, w, next;
//}e[10009];//边开小了!因为边是双向的！
}e[20009];
	void init()
{
	tot = 0;
	for (int i = 1; i <= N; i++) head[i] = -1;
	for (int i = 1; i <= N; i++) e[i].next = -1;
}
void addEdge(int u, int v, int w)
{
	e[tot].u = u;
	e[tot].v = v;
	e[tot].w = w;
	e[tot].next = head[u];
	head[u] = tot;
	tot++;
}

int flag[10009];//是否到达过，用于剪枝
int ans_v1[10009];//存v1答案
int ans_v2[10009];//存v2答案
int MAX;//记录最大长度
int v1, v2;//两个端点

void dfs_v0(int u)
{
	if (flag[u] == 1)return;

	flag[u] = 1;
	for (int i = head[u]; i!=-1; i = e[i].next)
	{
		int v = e[i].v;
		if (flag[v] != 1)
		{
			ans_v1[v] = ans_v1[u] + e[i].w;
			if (ans_v1[v] > MAX)
			{
				MAX = ans_v1[v];
				v1 = v;
			}
		}
		dfs_v0(v);
	}
}

void dfs_v1(int u)
{
	if (flag[u] == 1)return;

	flag[u] = 1;
	for (int i = head[u]; i != -1; i = e[i].next)
	{
		int v = e[i].v;
		if (flag[v] != 1)
		{
			ans_v1[v] = ans_v1[u] + e[i].w;
			if (ans_v1[v] > MAX)
			{
				MAX = ans_v1[v];
				v2 = v;
			}
		}
		dfs_v1(v);
	}
}

void dfs_v2(int u)
{
	if (flag[u] == 1)return;

	flag[u] = 1;
	for (int i = head[u]; i != -1; i = e[i].next)
	{
		int v = e[i].v;
		if (flag[v] != 1)
		{
			ans_v2[v] = ans_v2[u] + e[i].w;
			if (ans_v2[v] > MAX)
			{
				MAX = ans_v2[v];
				//v2 = v;这里不能再更新否则变为v1
			}
		}
		dfs_v2(v);
	}
}

int main()
{
	while (cin>>N)
	{
		memset(e, 0, sizeof(e));
		memset(flag, 0, sizeof(flag));
		memset(ans_v1,0,sizeof(ans_v1));
		memset(ans_v2,0,sizeof(ans_v2));
		init();

		int u, v, w;
		for (int i = 2; i <=N; i++)
		{//存图
			cin >> v >> w;
			addEdge(i, v, w);
			addEdge(v, i, w);
		}

		MAX = 0;
		dfs_v0(1);//找到v1
		MAX = 0;
		memset(flag, 0, sizeof(flag));
		memset(ans_v1, 0, sizeof(ans_v1));//记得重置第一遍的搜索结果，防止初值产生影响
		dfs_v1(v1);//找到v2并进行第一遍搜索
		MAX = 0;
		memset(flag, 0, sizeof(flag));
		dfs_v2(v2);//从v2进行第二遍搜索

		for (int i = 1; i <=N; i++)
			cout << max(ans_v1[i], ans_v2[i]) << endl;
	}
	return 0;
}