给定一个 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>usingnamespace std;constint N =1010;int n, m;int l[N], r[N], p[N];int dist[N];voiddfs(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);}intget_lca(int a,int b){if(dist[a]< dist[b])returnget_lca(b, a);while(dist[a]> dist[b]) a = p[a];while(a != b) a = p[a], b = p[b];return a;}intmain(){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;}}return0;}