第四次csp认证题4

做的时候有点搞心态，一直以为dfs返回的是u的最长路径，然后一直想不对，故提供一下自己的注释版，希望能帮到大家

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

using namespace std;

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

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

//dfs的目的是返回u的父节点到叶子结点的最长的一条路径
int dfs(int u){
    int d1=0,d2=0;//d1是u到叶子结点的最长路径，d2是u到叶子结点的次长路径
    for(int i=h[u];i!=-1;i=ne[i]){
        int j=e[i];
        int d=dfs(j);//求的是子节点j的叶子结点到父结点u的最长距离d
        //也就是返回u经过j这一个子节点的最长路径
        if(d>d1) d2=d1,d1=d;
        else if(d>d2) d2=d;
    }
    ans=max(ans,d1+d2);
    return d1+1;//返回的是u到叶子结点到u的父结点的最长的距离
    //u的到叶子结点的最长距离是d1，u到父节点的距离是1
}

int main()
{
    cin >> n >> m;
    memset(h, -1, sizeof h);
    for(int i=2;i<=n;i++){
        int p;
        cin >> p;
        add(p,i);
    }
    for(int i=n+1;i<=m+n;i++){
        int p;
        cin >> p;
        add(p,i);
    }
    
    dfs(1);
    
    cout<<ans;
    
    return 0;
}