hdu3848——树形dp

题意：求一棵树上任意两个叶子节点之间最短的路径。

对于那条我们要求的目标路径上的每一个节点，容易看出其到两个叶子节点的路径长度为其到所有叶子节点的最短的和次短的。如果我们在dfs的过程中，对于该路径中第一个到底的节点，我们计算出这个节点到所有的叶子节点最短和次短路径长度就是我们要求的解。我们在dfs的过程记录该节点到所有的叶子节点的最短和次短路径的长度。

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

const int maxn = 10000 + 100;
const int inf = 0x3f3f3f3f;
struct edge
{
	int from, to, dist;
};
vector<edge> edges;
vector<int> g[maxn];
int n, dp[maxn][2], dd[maxn], ans;

void addedge(int from, int to, int dist)
{
	edges.push_back((edge){from, to, dist});
	g[from].push_back(edges.size() - 1);
}

void prework()
{
	memset(dp, 0x3f, sizeof(dp));
	memset(dd, 0, sizeof(dd));
	edges.clear();
	for(int i = 1; i <= n; ++i) g[i].clear();
	for(int i = 0; i < n - 1; ++i)
	{
		int a, b, c;
		scanf("%d %d %d", &a, &b, &c);
		addedge(a, b, c);
		addedge(b, a, c);
		dd[a]++; dd[b]++;
	}
	ans = inf;
}

void dfs(int u, int fa)
{
	if(dd[u] == 1) dp[u][0] = 0;
	int s;
	for(int i = 0; i < g[u].size(); ++i)
	{
		edge & e = edges[g[u][i]];
		if(e.to == fa) continue;
		dfs(e.to, u);
		if(dp[u][0] > dp[e.to][0] + e.dist) dp[u][0] = dp[e.to][0] + e.dist, s = e.to;
	}
	for(int i = 0; i < g[u].size(); ++i)
	{
		edge & e = edges[g[u][i]];
		if(e.to == fa) continue;
		if(e.to != s) dp[u][1] = min(dp[u][1], dp[e.to][0] + e.dist);
		else dp[u][1] = min(dp[u][1], dp[e.to][1] + e.dist);
	}
	ans = min(ans, dp[u][0] + dp[u][1]);
}

void solve()
{
	dfs(1, -1);
	printf("%d\n", ans);
}

int main()
{
	while(~scanf("%d", &n) && n)
	{
		prework();
		solve();
	}
	return 0;
}