[SGU - 149 Computer Network] 树形DP
题意描述:给定N个顶点的一棵带权树,求树上每个节点的最长路长度。N<=10000
解题思路:很不错的一道树形DP!
对根节点 rt 来一次DFS肯定是可以求出根节点的树上最长路。但是,并不能求出所有的答案。
假如树上任意一个节点 u, 求它的一个子节点 v 的树上最长路。 我们可以分两种情况讨论:
- v 在 u 的最长路的路径上面:那么v 的最长路就等于 max{ 从节点u出发能够到达的最长路 - |(u,v)|, 从节点u出发能够到达的次长路 + |(u,v)| }。
- v不在 u 的最长路的路径上面:那么v的最长路就等于 max{ 从节点u出发能够到达的最长路 + |(u,v)|, 从节点v出发能够到达的最长路 }。
所以,我们需要求出
根节点最长路和次长路,然后利用上面的状态转移,求出子节点的最长路和次长路。
求根节点最长路和次长路一次DFS可以搞定。然后再用一次
前序DFS从根节点出发,依次更新子节点的最长路和次长路。
#include <cmath>
#include <queue>
#include <vector>
#include <cstdio>
#include <string>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <algorithm>
using namespace std;
//#pragma comment(linker, "/STACK:1024000000,1024000000")
#define FIN freopen("input.txt","r",stdin)
#define FOUT freopen("output.txt","w",stdout)
#define fst first
#define snd second
typedef __int64 LL;
//typedef long long LL;
typedef unsigned int uint;
typedef pair<int, int> PII;
const int INF = 0x3f3f3f3f;
const double eps = 1e-6;
const int MAXN = 10000 + 5;
struct Edge {
int v, w, next;
Edge() {}
Edge (int v, int w, int next) : v (v), w (w), next (next) {}
} edges[MAXN << 1];
int head[MAXN], tot;
int N;
int dist[MAXN][2];
void init_graph () {
tot = 0;
memset (head, -1, sizeof (head) );
}
void add_edge (int u, int v, int w) {
edges[tot] = Edge (v, w, head[u]);
head[u] = tot ++;
}
void dfs1 (int u, int pre) {
int v, w;
for (int i = head[u]; ~i; i = edges[i].next) {
v = edges[i].v, w = edges[i].w;
if (v == pre) continue;
dfs1 (v, u);
if (dist[u][1] < dist[v][1] + w) {
dist[u][0] = dist[u][1];
dist[u][1] = dist[v][1] + w;
} else {
dist[u][0] = max (dist[u][0], dist[v][1] + w);
}
}
}
void dfs2 (int u, int pre) {
int v, w;
for (int i = head[u]; ~i; i = edges[i].next) {
v = edges[i].v, w = edges[i].w;
if (v == pre) continue;
if (dist[u][1] == dist[v][1] + w) {
dist[v][0] = max (dist[v][0], min (dist[u][1] - w, dist[u][0] + w) );
dist[v][1] = max (dist[u][1] - w, dist[u][0] + w);
} else {
dist[v][0] = max (dist[v][0], min (dist[u][1] + w, dist[v][1]) );
dist[v][1] = max (dist[u][1] + w, dist[v][1]);
}
dfs2 (v, u);
}
}
int main() {
#ifndef ONLINE_JUDGE
FIN;
#endif // ONLINE_JUDGE
int u, v, w, rt;
while (~scanf ("%d", &N) ) {
init_graph();
for (u = 2; u <= N; u++) {
scanf ("%d %d", &v, &w);
add_edge (u, v, w);
add_edge (v, u, w);
}
rt = 1;
memset (dist, 0x00, sizeof (dist) );
dfs1 (rt, -1);
dfs2 (rt, -1);
for (int i = 1; i <= N; i++) {
printf ("%d\n", dist[i][1]);
}
}
return 0;
}