二叉树暑假每日一题 34

给定一个 $n$ 个结点（编号 $1\sim n$）构成的二叉树，其根结点为 $1$ 号点。

进行 $m$ 次询问，每次询问两个结点之间的最短路径长度。

树中所有边长均为 $1$。

输入格式
第一行包含一个整数 $T$，表示共有 $T$ 组测试数据。

每组数据第一行包含两个整数 $n,m$。

接下来 $n$ 行，每行包含两个整数，其中第 $i$ 行的整数表示结点 $i$ 的子结点编号。如果没有子结点则输出 −1。

接下来 $m$ 行，每行包含两个整数，表示要询问的两个结点的编号。

输出格式
每组测试数据输出 $m$ 行，代表查询的两个结点之间的最短路径长度。

数据范围
$1\le T\le 10,$
$1\le n,m\le 1000$

输入样例：

1
8 4
2 3
4 5
6 -1
-1 -1
-1 7
-1 -1
8 -1
-1 -1
1 6
4 6
4 5
8 1

输出样例：

2
4
2
4

---

#include<iostream>

using namespace std;

const int N = 1010;

int n, m;
int d[N], p[N];
int l[N], r[N];

void dfs(int u, int depth){
    
    d[u] = depth;
    if(l[u] != -1) dfs(l[u], depth + 1);
    if(r[u] != -1) dfs(r[u], depth + 1);
}

int get(int x, int y){
    
    int res = 0;
    while(x != y){
        res++;
        if(d[x] > d[y]) x = p[x];
        else y = p[y];
    }
    return res;
}

int main(){
    
    int t;
    cin >> t;
    while(t--){
        
        cin >> n >> m;
        for(int i = 1; i <= n; i++){
            cin >> l[i] >> r[i];
            if(l[i] != -1) p[l[i]] = i;
            if(r[i] != -1) p[r[i]] = i;
        }
        
        dfs(1, 1);
        
        while(m--){
            int x, y;
            cin >> x >> y;
            cout << get(x, y) << endl;
        }
    }
    
    return 0;
}