树的直径 无边权改编

#include<bits/stdc++.h>

using namespace std;

const int N = 2e4 + 50;

const int M = 2e4 + 50;

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 != - 1; 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;
}

int main()
{
    scanf("%d %d", &n, &m);
    memset(h, -1, sizeof h);

    for(int i = 2; i <= n; i++)
    {
        int p;
        scanf("%d", &p);
        add(p, i);
    }

    for(int i = n + 1; i <= n + m; i++)
    {
        int p;
        scanf("%d", &p);
        add(p, i);
    }

    dfs(1);
    printf("%d", ans);
    return 0;
}