二叉树(北京邮电大学机试题）（ＤＡＹ ８５）

1：题目

给定一个 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


难度：简单
时/空限制：1s / 64MB
总通过数：654
总尝试数：1453
来源：北京邮电大学考研机试题
算法标签

2：代码实现

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1010;

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

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

int get_lca(int a, int b)
{
    if (dist[a] < dist[b]) return get_lca(b, a);
    while (dist[a] > dist[b]) a = p[a];
    while (a != b) a = p[a], b = p[b];
    return a;
}

int main()
{
    int T;
    cin >> T;
    while (T -- )
    {
        cin >> n >> m;
        for (int i = 1; i <= n; i ++ )
        {
            int a, b;
            cin >> a >> b;
            l[i] = a, r[i] = b;
            if (a != -1) p[a] = i;
            if (b != -1) p[b] = i;
        }

        dfs(1, 0);

        while (m -- )
        {
            int a, b;
            cin >> a >> b;
            int c = get_lca(a, b);
            cout << dist[a] + dist[b] - dist[c] * 2 << endl;
        }

    }

    return 0;
}