算法

求树的直径问题，可以使用一次DFS

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 20010, M = N;

int n, m;
int h[N], e[M], ne[M], idx;
int ans;

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

int dfs(int u) {
    int d1 = 0, d2 = 0; //最大值与次大值
    for (int i = h[u]; ~i; i = ne[i]) {
        int j = e[i]; //当前点的编号
        int d = dfs(j); //当前点子树的最大距离
        if (d >= d1) d2 = d1, d1 = d;
        else if (d > d2) d2 = d;
    }
    ans = max(ans, d1 + d2);
    return d1 + 1; //+1是返回子树根节点到父节点的边
}

int main() {
    scanf("%d%d", &n, &m);
    memset(h, -1, sizeof h);
    for (int i = 2; i <= n; i ++ ) {
        int a;
        scanf("%d", &a);
        add(a, i);
    }
    for (int i = n + 1; i <= n + m; i ++ ) {
        int b;
        scanf("%d", &b);
        add(b, i);
    }
    dfs(1);
    printf("%d\n", ans);
    return 0;
}

【CCF-CSP】网络延时