算法-程序设计课week6-作业-A - 氪金带东

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，每个数之间用一个空格隔开。

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，其最远点一定是端点其中的一个，因此我们只需要求出端点编号，计算两个端点到i的距离，然后在i的两个距离中选最大的那个。

1. 读取输入为无向图
2. 一次dfs，从任意一个点开始，求他的最远点，这个点必定是两端点中的一个。
3. 第二次dfs，从上一次的最远点开始，遍历所有节点，求出每个节点到此最远点的距离，顺便求出第二个端点。
4. 第三次dfs，从第二个最远点开始，同上进行遍历，选取点i到两个端点间较大的那个作为答案。

心得

这个题心算看起来挺简单，但要设计较优的算法，还是比较困难的。

1. 首先要考虑时间复杂度，如果对每个点都用一遍dfs，势必造成超时，因此要将路径计算过程复用。
2. 然后要考虑空间复杂度，如果采用二维数组来保存节点间的联系，内存是不够的，因此采用稀疏矩阵，只记录相连节点的信息。

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

int end_point;		  //记录端点编号
int max_len;		  //记录整个图的最长路径，它一直是唯一的
int max_dis[100010];  //保存i到两端点的最远距离

//图的节点
struct node {
	int toid;
	int distance;
	node(int a, int b) : toid(a), distance(b) {}
};

//图的矩阵
vector<vector<node> > matrx;

void dfs(int start, int from, int len) {
	if (max_len < len) {
		max_len = len;
		end_point = start;
	}

	//这里i只是索引，不代表具体意义
	for (int i = 0; i < matrx[start].size(); i++) {
		if (from == matrx[start][i].toid) continue;	 //代替了visited数组
		dfs(matrx[start][i].toid, start, len + matrx[start][i].distance);

		//选取最远者
		max_dis[matrx[start][i].toid] = max(len + matrx[start][i].distance,
											max_dis[matrx[start][i].toid]);
	}
}
int main() {
	matrx.resize(100010);

	int n;
	while (~scanf("%d", &n)) {
		max_len = 0;
		memset(max_dis, 0, sizeof(max_dis));

		//获取有向图连接信息
		for (int i = 2; i <= n; i++) {
			int num, len;
			scanf("%d %d", &num, &len);
			matrx[i].push_back(node(num, len));
			matrx[num].push_back(node(i, len));
		}

		//进行三次dfs
		dfs(1, -1, 0);			//获取第一个端点
		dfs(end_point, -1, 0);	//根据上次遍历获得的端点进行计算
		dfs(end_point, -1, 0);	//根据上次遍历获取的端点进行计算
		for (int i = 1; i <= n; i++) {
			printf("%d\n", max_dis[i]);
		}
	}
	return 0;
}