HDU2196——Computer(树形DP，经典)

题目连接

题目解析

经典的树形DP题。
题意是求树中每个点到所有叶子节点的距离的最大值是多少。
第一次DFS求出所有节点在他的子树范围内到叶子节点距离的最大值和第二大的值，第二次DFS更新从父节点过来的情况就可以了。
因为如果只存最大值的话，判断一个点的从父节点过来的最大值，那么如果他的父节点存的最大值正好是从该点过来的，那么就失去了从父节点过来的状态，所以要记录最大的两个值。

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

typedef long long LL;
const int MOD = 1e9 + 7;
const int INF = 0x7fffffff;
const int N = 1e4 + 10;

vector <pair<int, int> > G[N];
int fid[N], sid[N], fdis[N], sdis[N];

void dfs1(int u, int fa) {
    for(int i = 0; i < G[u].size(); i++) {
        int v = G[u][i].first, w = G[u][i].second;
        if(v == fa) continue;
        dfs1(v, u);
        int d = fdis[v] + w;
        if(d > fdis[u]) {
            sdis[u] = fdis[u], fdis[u] = d;
            sid[u] = fid[u], fid[u] = v;
        } else if(d > sdis[u]) {
            sdis[u] = d;
            sid[u] = v;
        }
    }
}
void dfs2(int u, int fa) {
    for(int i = 0; i < G[u].size(); i++) {
        int v = G[u][i].first, w = G[u][i].second;
        if(v == fa) continue;
        if(v == fid[u]) {
            if(fdis[v] < sdis[u] + w) {
                sdis[v] = fdis[v], fdis[v] = sdis[u] + w;
                sid[v] = fid[v], fid[v] = u;
            } else if(sdis[v] < sdis[u] + w) {
                sdis[v] = sdis[u] + w;
                sid[v] = u;
            }
        } else {
            if(fdis[v] < fdis[u] + w) {
                sdis[v] = fdis[v], fdis[v] = fdis[u] + w;
                sid[v] = fid[v], fid[v] = u;
            } else if(sdis[v] < fdis[u] + w) {
                sdis[v] = fdis[u] + w;
                sid[v] = u;
            }
        }
        dfs2(v, u);
    }
}

int main() {
#ifdef TYH
    freopen("in.txt", "r", stdin);
#endif // TYH
    int n;
    while(scanf("%d", &n) != EOF) {
        for(int i = 1; i <= n; i++) {
            fid[i] = sid[i] = fdis[i] = sdis[i] = 0;
            G[i].clear();
        }
        for(int x = 2; x <= n; x++) {
            int y, w;
            scanf("%d%d", &y, &w);
            G[x].push_back(make_pair(y, w));
            G[y].push_back(make_pair(x, w));
        }
        dfs1(1, 0);
        dfs2(1, 0);
        for(int i = 1; i <= n; i++) {
            printf("%d\n", fdis[i]);
        }
    }
    return 0;
}