天梯赛 L2-043 龙龙送外卖

 深搜，距离的2倍减去最大深度，因为不用返回起点

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int ans;
int fa[N];//节点的父亲节点
int n,m;
bool vis[N];//是否访问过该节点
int maxv;//从起点到各个点中的最大深度
int t;//记录在dfs函数中的d
int dis[N];//记录每个节点的深度
void dfs(int node,int d)
{
    if(vis[node])
    {
        maxv=max(maxv,dis[node]+d);
        t=d;//不是深度，是从距离该节点最近的已访问的父亲节点的深度
        return;
    }
    dfs(fa[node],d+1);
    vis[node]=true;
    dis[node]=dis[fa[node]]+1;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
      cin>>fa[i];
     if(fa[i]==-1)
     {
         fa[i]=i;
         vis[i]=true;//起点标记为已访问
     }
    }
    for(int i=0;i<m;i++)
    {
        int a;
        cin>>a;
        int d=0;//初始化
        dfs(a,d);
        ans+=t*2;//因为是是来回，所以要2倍
        int sum=ans;
        sum-=maxv;
        cout<<sum<<endl;
    }
    return 0;
}