求树的直径例题

分析

本题其实是让我们求树的直径。
由于本题有n+m-1个节点，所以N要赋值为2e4+10。
之后利用树形Dp思想求树的直径。

C++ 代码

#include<bits/stdc++.h>
using namespace std;
const int N = 2e4+10;   //注意是n+m-1，要开2e4，不然会MLE
int n,m,ans;
int h[N],e[2*N],ne[2*N],idx;
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u,int father)   //u为当前点，father为其父节点
{
    int dist=0;     //从当前点向下走的最大长度
    int d1=0,d2=0;  //d1为经过当前点的最大长度，d2为第二大长度
    for(int i=h[u];~i;i=ne[i])
    {
        int j=e[i];
        if(j==father) continue;
        int d=dfs(j,u)+1;
        dist=max(dist,d);
        if(d>=d1) d2=d1,d1=d;
        else if(d>=d2) d2=d;
    }
    ans=max(ans,d1+d2); 
    return dist;    //返回从当前点向下走的最大长度
}
int main()
{
    memset(h,-1,sizeof h);
    cin>>n>>m;
    int a;
    for(int i=2;i<=n;i++)
    {
        cin>>a;
        add(a,i),add(i,a);
    }
    for(int i=1+n;i<=m+n+1;i++) //注意电脑的编号应该为n+1~n+m+1，不然会和交换机编号重合
    {
        cin>>a;
        add(a,i),add(i,a);
    }
    dfs(1,-1);
    cout<<ans;
    return 0;
}