【CCF】网络延时 树搜索

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>

using namespace std;
const int maxn=2e4+;
const int maxm=2e4+;
const int inf=0x3f3f3f3f;
int n,m;
struct edge{
    int to;
    int nxt;
}e[*maxm];
int len[maxn][];
bool vis[maxn];
int tot;
int head[maxn];
int ans=-inf;
void init(){
    memset(head,-,sizeof(head));
    tot=;
    memset(len,,sizeof(len));
    memset(vis,false,sizeof(vis));
}
void add(int u,int v){
    e[tot].to=v;
    e[tot].nxt=head[u];
    head[u]=tot++;
}
void dfs(int u){
    vis[u]=true;
    vector<int> vec;
    for(int i=head[u];i!=-;i=e[i].nxt){
        int v=e[i].to;
        if(vis[v]) continue;
        dfs(v);
        int tmp=len[v][]+;
        vec.push_back(tmp);
    }
    int sz=vec.size();
    if(sz==){
        len[u][]=len[u][]=;
    }else if(sz==){
        len[u][]=vec[];
        len[u][]=;
    }else{
        int pos1=-,pos2=-;
        for(int i=;i<sz;i++){
            if(vec[i]>len[u][]){
                len[u][]=vec[i];
                pos1=i;
            }
        }
        for(int i=;i<sz;i++){
            if(i==pos1) continue;
            if(vec[i]>len[u][]){
                len[u][]=vec[i];
            }
        }
        ans=max(ans,len[u][]+len[u][]);
    }
}
int work(){
    dfs();
    return ans;
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        init();
        int x;
        for(int i=;i<=n;i++){
            scanf("%d",&x);
            add(i,x);
            add(x,i);
        }
        for(int i=;i<=m;i++){
            scanf("%d",&x);
            add(n+i,x);
            add(x,n+i);
        }
        int ans=work();
        printf("%d\n",ans);
    }
    return ;
}