HDU 2196 Computer（树上最远距离）

题目链接：

Computer

 

题意：

有一棵树，树上每条边都有一个权值，求每个点与其距离最远的点之间的距离。

 

思路：

dp[root][state]：以root为根节点的子树，root的状态为state的情况下的最远距离。

dp[root][0]：root在以root为根节点的子树中的最远距离。

dp[root][1]：root在以root为根节点的子树中的次远距离(不是距离值第二长，而是指除去取到最远距离的那个子节点后的最远距离)。

dp[root][2]：root的父节点不经过root所能取到的最远距离 + root到其父节点的距离。

首先，思考一个点取到与其距离最远的点之间的距离的情况种类：

一：在其子树中找到。

二：往其父节点那个方向找。( 找在删去其子树的树中，离其父节点最远的距离 )

解：

首先一次dfs1能计算完所有的dp[root][0]。此外对于dp[root][1]，就是在root算完dp[root][0]后，把取到最大值的那个子节点除掉后的最远距离，因此可以用一个数组bigson[root]表示root在其子树中取到最大值经过的那个与其直接相连的子节点，然后碰到这个点就跳过，用和计算dp[root][0]一样的方法就能得到dp[root][1]。

然后再用一次dfs2来计算dp[root][2]。可知要么往root父节点的子树中找，要么往其父节点的父节点方向去找。

有2种情况：设v为root的子节点，val为root到v这条边的距离

如果root取到其子树中的最大值是经过v的，那么dp[root][2]=max(dp[root][1],dp[root][2])+val。

否则，dp[root][2]=max(dp[root][0],dp[root][2])+val。

dfs1更新了所有子树的特点，子树的信息可以直接用了，父节点的信息从一步步dfs2下去也都已经更新好了，也可以直接用的，因此一直更新最优。

 

Code：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

const int MAX = 10000 + 10;
const int inf = 1e9 + 7;

typedef struct {
	int to, val;
}Point;

int n;
vector<Point>mp[MAX];
int dp[MAX][3];
int bigson[MAX];

void dfs1(int root, int fa)
{
	for (int i = 0; i < mp[root].size(); i++) {
		int v = mp[root][i].to;
		int w = mp[root][i].val;
		if (v == fa)	continue;
		dfs1(v, root);
		if (dp[root][0] < dp[v][0] + w) {
			dp[root][0] = dp[v][0] + w;
			bigson[root] = v;
		}
	}
	for (int i = 0; i < mp[root].size(); i++) {
		int v = mp[root][i].to;
		int w = mp[root][i].val;
		if (v == fa)	continue;
		if (v == bigson[root])	continue;
		dfs1(v, root);
		dp[root][1] = max(dp[root][1], dp[v][0] + w);
	}
}

void dfs2(int root, int fa)
{
	for (int i = 0; i < mp[root].size(); i++) {
		int v = mp[root][i].to;
		int w = mp[root][i].val;
		if (v == fa)	continue;
		if (v == bigson[root]) {
			dp[v][2] = max(dp[root][1], dp[root][2]) + w;
		}
		else {
			dp[v][2] = max(dp[root][0], dp[root][2]) + w;
		}
		dfs2(v, root);
	}
}

int main()
{
	while (scanf("%d", &n) != EOF)
	{
		memset(dp, 0, sizeof(dp));
		memset(bigson, 0, sizeof(bigson));
		for (int i = 1; i <= n; i++)	mp[i].clear();
		for (int i = 2; i <= n; i++) {
			int u, w;
			scanf("%d%d", &u, &w);
			mp[i].push_back(Point{ u,w });
			mp[u].push_back(Point{ i,w });
		}
		dfs1(1, -1);
		dfs2(1, -1);
		for (int i = 1; i <= n; i++) {
			printf("%d\n", max(dp[i][0], dp[i][2]));
		}
	}
	return 0;
}