AcWing 3215. 网络延时（树的直径 + BFS + DFS + 树形DP）

题目链接

给定一个公司的网络，由 n 台交换机和 m 台终端电脑组成，交换机与交换机、交换机与电脑之间使用网络连接。

交换机按层级设置，编号为 1 的交换机为根交换机，层级为 1。

其他的交换机都连接到一台比自己上一层的交换机上，其层级为对应交换机的层级加 1。

所有的终端电脑都直接连接到交换机上。

当信息在电脑、交换机之间传递时，每一步只能通过自己传递到自己所连接的另一台电脑或交换机。

请问，电脑与电脑之间传递消息、或者电脑与交换机之间传递消息、或者交换机与交换机之间传递消息最多需要多少步。

输入格式

输入的第一行包含两个整数 n,m，分别表示交换机的台数和终端电脑的台数。

第二行包含 n−1 个整数，分别表示第 2、3、……、n 台交换机所连接的比自己上一层的交换机的编号。第 i 台交换机所连接的上一层的交换机编号一定比自己的编号小。

第三行包含 m 个整数，分别表示第 1、2、……、m 台终端电脑所连接的交换机的编号。

输出格式

输出一个整数，表示消息传递最多需要的步数。

数据范围

前 30% 的评测用例满足：n≤5,m≤5。
前 50% 的评测用例满足：n≤20,m≤20。
前 70% 的评测用例满足：n≤100,m≤100。
所有评测用例都满足：1≤n≤10000，1≤m≤10000。

输入样例1：

4 2
1 1 3
2 1

输出样例1：

4

样例1解释

样例的网络连接模式如下，其中圆圈表示交换机，方框表示电脑：

其中电脑 1 与交换机 4 之间的消息传递花费的时间最长，为 4 个单位时间。

输入样例2：

4 4
1 2 2
3 4 4 4

输出样例2：

4

样例2解释

样例的网络连接模式如下：

其中电脑 1 与电脑 4 之间的消息传递花费的时间最长，为 4 个单位时间。

思路：

树的最长路径模板题

答案：

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f3f3f3f
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-6
const int mod = 1e9 + 7;
const int MOD = 1e4+7;
const int N = 2e6 + 10;
const int M = 1111;
int dx[]={-1, 0, 1, 0};
int dy[]={0, 1, 0, -1};
int dxy[][2]={{0,1},{1,0},{1,1},{-1,1}};
using namespace std;

int e[N] , ne[N] , h[N] , idx;
int f[N];
int n , m , ans;

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

void DFS(int u){
    int a = 0 , b = 0;
    for(int i = h[u] ; ~i ; i = ne[i]){
        int j = e[i];
        DFS(j);
        int t = f[j] + 1;
        if(t >= a)
            b = a , a = t;
        else if(t > b)
            b = t;
    }
    f[u] = a;
    ans = max(ans , a + b);
}

void solve(){
    cin >> n >> m;
    memset(h , -1 , sizeof h);
    for(int i = 2 ; i <= n + m ; i++){
        int j;
        cin >> j;
        add(j , i);
    }
    DFS(1);
    cout << ans << endl;
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    solve();
	return 0;
}