HDU

传送门

求从树上每一点出发能得到的最长路。

对于一个结点node，从它出发的最长路要么是它向子树走能得到的最大值，要么是先走向parent，再加上parent的不经过node的最长路的值。此时parent有可能走向它的父亲，或者走向node的兄弟结点。所以每个点我们要记录向子树走的最长路dp[node][1]和次长路dp[node][2]，还要记录经过其父亲结点的最长路dp[node][3]。

前两个可以在一次dfs中处理出，最后再用一次dfs求出经过父亲的最长路

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

const int maxn = 1e4 + 10;
int N;
int dp[maxn][4];
vector<int> tree[maxn];

void dfs1(int par) {
    int fir = 0, sec = 0;
    for (int i = 0; i < tree[par].size(); i++) {
        int v = tree[par][i];
        dfs1(v);
        int tmp = dp[v][1] + dp[v][0];
        if (tmp >= fir) {
            sec = fir; fir = tmp;
        } else if (tmp > sec) {
            sec = tmp;
        }
    }
    dp[par][1] = fir; dp[par][2] = sec;
}

void dfs2(int par) {
    for (int i = 0; i < tree[par].size(); i++) {
        int v = tree[par][i];
        dp[v][3] = dp[v][0] + max(dp[par][3],
                                    (dp[v][0] + dp[v][1] == dp[par][1] ? dp[par][2] : dp[par][1]));
        dfs2(v);
    }
}

int main() {
    while (~scanf("%d", &N)) {
        memset(dp, 0, sizeof(dp));
        for (int i = 0; i <= N; i++) tree[i].clear();
        for (int i = 2; i <= N; i++) {
            int u;
            scanf("%d%d", &u, &dp[i][0]);
            tree[u].push_back(i);
        }
        dfs1(1);
        dfs2(1);
        for (int i = 1; i <= N; i++) printf("%d
", max(dp[i][1],dp[i][3]));
    }
    return 0;
}