二叉树（暑假每日一题 34）

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

进行 $m$

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

输入格式
第一行包含一个整数 $T$，表示共有 $T$

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

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

接下来 $m$

输出格式
每组测试数据输出 $m$

数据范围
$1≤T≤10,$
$1≤n,m≤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;
}