Week6—A—氪金带东

题目描述：

实验室里原先有一台电脑(编号为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，每个数之间用一个空格隔开。

sample input：

	5
	1 1
	2 1
	3 1
	1 1

Output：

对于每组测试数据输出N行，第i行表示i号电脑的答案 (1<=i<=N).

sample output：

	3
	2
	3
	4
	4

---

个人思路：

简单来说就是跑了四遍DFS，

1、第一遍找到直径的一个端点s
2、第二遍找到直径的另一个端点t
3、第三遍找到从s到树中节点x的距离
4、第四遍找到从t到树中节点x的距离

简单粗暴，理论上可以简化成3次DFS，那就是在第二遍的时候记录第三遍的距离；不过为了以后能看懂写的啥， 这样说比较容易理解。

---

代码块：

图的存储用的是邻接表。

#include<iostream>
#include<queue>
using namespace std;
const int maxn = 10100;
int n, s, t;
long long dis = 0;
long long now_dis = 0;
long long disn[maxn] = { 0 };
struct Edge {
	int u, v;
	long long w;
	Edge(int u1, int v1, long long w1) {
		u = u1;
		v = v1;
		w = w1;
	}
};
vector<Edge> G[maxn];
void addedge(int u, int v, long long w) {
	G[u].push_back(Edge(u, v, w));
	G[v].push_back(Edge(v, u, w));
	//双向
}
void initialEdge() {
	memset(disn, 0, sizeof(disn));
	dis = 0;
	for (int i = 1; i <= n; ++i) {
		G[i].clear();
	}
}
void DFS_s(int u,int last) {
	for (int i = 0; i < G[u].size(); ++i) {
		if (G[u][i].v == last)
			continue;
		now_dis += G[u][i].w;
		DFS_s(G[u][i].v,u);
		if (dis <= now_dis) { dis = now_dis; s = G[u][i].v; disn[s] = dis; }
		now_dis -= G[u][i].w;
	}
}
void DFS_t(int u, int last) {
	for (int i = 0; i < G[u].size(); ++i) {
		if (G[u][i].v == last)
			continue;
		now_dis += G[u][i].w;
		DFS_t(G[u][i].v, u);
		if (dis <= now_dis) { dis = now_dis; t = G[u][i].v; disn[t] = dis; }
		now_dis -= G[u][i].w;
	}
}
void DFS_s_x(int u, int last) {
	for (int i = 0; i < G[u].size(); ++i) {
		if (G[u][i].v == last)
			continue;
		now_dis += G[u][i].w;
		DFS_s_x(G[u][i].v, u);
		disn[G[u][i].v] = now_dis;
		now_dis -= G[u][i].w;
	}
}
void DFS_t_x(int u, int last) {
	for (int i = 0; i < G[u].size(); ++i) {
		if (G[u][i].v == last)
			continue;
		now_dis += G[u][i].w;
		DFS_t_x(G[u][i].v, u);
		disn[G[u][i].v] = max(now_dis, disn[G[u][i].v]);
		now_dis -= G[u][i].w;
	}
}
int main() {
	while (~scanf_s("%d", &n)) {
		initialEdge();
		int v; long long w;
		for (int i = 2; i <= n; ++i) {
			scanf_s("%d%lld", &v, &w);
			addedge(i, v, w);
		}
		DFS_s(1, 1);
		dis = 0;
		DFS_t(s, s);
		DFS_s_x(s, s);
		DFS_t_x(t, t);
		for (int i = 1; i <= n; ++i) {
			printf("%lld\n", disn[i]);
		}
	}
	return 0;
}

---