求二叉树中两点的距离【LCA最近公共祖先写法 + 思路分析 + 代码注释】

思路：

对于求二叉树中两点的距离，我们可以利用二叉树的特性，即采用最近公共祖先的特性。因为二叉树中的任意一个点在二叉树中都是有祖先的。并且任意两个点必定具有公共祖先，最坏情况也就是两个叶子节点的公共祖先是根节点而已，但是也算公共祖先。

故我们可以画图推导公式如下：
假设第i个点到根节点的距离是：dist[i]; 那么我们要求4 和 5号节点之间的距离，不难发现她们都具有相同的一段距离，即为2—>1这段距离。
(dist[5] - dist[2]) + (dist[4] - dist[2]) = dist[5] + dist[4] - 2*dist[2]，即可！
所以第一步是预处理出来每个节点到根节点的距离，即dist数组。
然后第二步就是查询的时候找出我们的两个点的最近公共祖先。

第一步：求dist数组：
采用递归的方式求解，只要当前节点不是空节点，则每次往下走一步，距离+1；

第二步：求最近公共祖先：
向上爬坡法求解：先将其中深度更高的点爬，爬到与深度低的点同等高度后，再二者同时向上爬坡，直到爬到相同节点为止。

另外一种爬法，就是先让一个节点一直向上爬，爬到根节点为止，然后每爬一个节点就存入集合中。然后另一个节点再爬，边爬边判断。

时间复杂度为 O(n)；

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

using namespace std;

const int N = 1e3 + 10;
int n, m;
int l[N], r[N], p[N];   //分别表示i的左儿子，i的右儿子，i的父节点是谁。
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 query(int a, int b) //查询两个人的最近公共祖先！
{
    //可以利用a和b到达根节点的距离，从而判断哪个点的深度更大！
    if (dist[a] > dist[b]) swap(a, b);  //默认a的深度比b的深度更小，即b在a上面，这样做简化代码，从而让a先爬
    while (dist[a] < dist[b]) b = p[b]; //b在下
    
    //此时弹出上述循环后，dist[a] == dist[b];
    while (a != b)// 然后让他们一起爬，爬到同一个点为止！
    {
        a = p[a];
        b = p[b];
    }
    
    return a;
}

int main()
{
    int t;
    cin >> t;
    while (t --)
    {
        cin >> n >> m;
        memset (l, -1, sizeof (l));
        memset (r, -1, sizeof (r));
        memset (dist, 0, sizeof (dist));
        for (int i=1; i <= n; i ++)
        {
            int a, b;   //输入i号点的左右儿子节点！
            cin >>  a >> b; 
            l[i] = a, r[i] = b;
            if (a != -1) p[a] = i;  //如果a和b不为空，则记录他们的父节点！
            if (b != -1) p[b] = i;
        }
        //求dist数组！
        dfs (1, 0); //从根节点开始搜索！
        while (m -- )
        {
            int a, b;
            cin >> a >> b;  //输入待查询的点！
            int ancestor = query(a, b);
            cout << dist[a]+dist[b] - 2*dist[ancestor] << endl; //减去的是公共祖先，所以现在求的是公共祖先是谁！
        }
    }
    return 0;
}```