[SGU - 149 Computer Network] 树形DP 求带权树上每个节点的最长路长度

[SGU - 149 Computer Network] 树形DP

题意链接: [SGU - 149 Computer Network] 

题意描述：给定N个顶点的一棵带权树，求树上每个节点的最长路长度。N<=10000

解题思路：很不错的一道树形DP！

对根节点 rt 来一次DFS肯定是可以求出根节点的树上最长路。但是，并不能求出所有的答案。

假如树上任意一个节点 u， 求它的一个子节点 v 的树上最长路。 我们可以分两种情况讨论：

1.  v    在 u 的最长路的路径上面：那么v 的最长路就等于 max{ 从节点u出发能够到达的最长路 - |(u,v)|, 从节点u出发能够到达的次长路 + |(u,v)| }。 
2.  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;
}