201503-4 网络延时

树的直径。
板子

#include<bits/stdc++.h>
using namespace std;
vector<int>edge[100005];
int vis[100005];
int dis[100005];
void dfs(int st)
{
    for(int i=0;i<edge[st].size();i++)
    {
        int to=edge[st][i];
        if(!vis[to])
        {
            vis[to]=1;
            dis[to]=dis[st]+1;//注意，本代码计算的是无权树的直径，所以边权为1
            //如果是有权树，则这里的1要改为边权
            dfs(to);
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    int n;
    cin>>n;//n个点
    for(int i=1;i<=n-1;i++)//因为是树，有n-1条边
    {
        int u,v;
        cin>>u>>v;
        edge[u].push_back(v);
        edge[v].push_back(u);//无向图存储，若是有权树还要用结构体
    }
    vis[1]=1;
    dfs(1);//从1出发dfs一边
    int maxlen=0,Q,W,ans=0;
    for(int i=1;i<=n;i++)
    {
        if(dis[i]>maxlen) maxlen=dis[i],Q=i;
        dis[i]=vis[i]=0;
    }//找到直径的一个端点Q
    vis[Q]=1;
    dfs(Q);//从Q出发
    for(int i=1;i<=n;i++)
        if(dis[i]>ans) ans=dis[i],W=i;//找到另一个端点W，同时得到直径长度
    return 0;
}
参考：https://zhuanlan.zhihu.com/p/115966044?from_voters_page=true

这道题的代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=2e4+5,INF=0x3f;
int n,m;
vector<int>g[maxn];
int dis[maxn],vis[maxn];
void dfs(int u)
{
    for(int i=0;i<g[u].size();i++)
        if(!vis[g[u][i]])
        {
            dis[g[u][i]]=dis[u]+1;
            vis[g[u][i]]=1;
            dfs(g[u][i]);
        }
}
int main()
{
    cin>>n>>m;
    for(int i=2;i<=n;i++)
    {
        int u;
        cin>>u;
        g[u].push_back(i);
        g[i].push_back(u);
    }
    for(int i=n+1;i<=n+m;i++)
    {
        int u;
        cin>>u;
        g[u].push_back(i);
        g[i].push_back(u);
    }
    vis[1]=1;
    dfs(1);
    int ans=0,sum=0,q=0,p=0;
    for(int i=1;i<=n+m;i++)
        {
            //cout<<dis[i]<<endl;
            if(sum<dis[i]) {sum=dis[i];q=i;}
            //cout<<sum<<endl;
        }
    memset(dis,0,sizeof(dis));
    memset(vis,0,sizeof(vis));
    //cout<<q<<endl;
    vis[q]=1;
    dfs(q);
    for(int i=1;i<=n+m;i++)
        {
            if(ans<dis[i]) {ans=dis[i];p=i;}
            //cout<<dis[i]<<endl;
        }
    //cout<<p<<endl;
    cout<<ans<<endl;
    return 0;
}