二叉树两个结点之间的最短距离

二叉树两个结点之间的最短距离

step1 建立树
step2 结点到根的路径
step3 找交点 算距离

#include <iostream>
#include <vector>
using namespace std;
struct TreeNode{//树结点
    int num;
    TreeNode* left;
    TreeNode* right;
    TreeNode* parent;
};
int main(){
    int t;
    int n,m;
    scanf("%d",&t);
    for(int i=0;i<t;++i){
        scanf("%d%d",&n,&m);
        vector<TreeNode*> nodeArr(n+1);
        for(int j=1;j<=n;++j){
            nodeArr[j]=new TreeNode;
            nodeArr[j]->num=j;
        }//动态数组nodeArr存放每个树的结点
        nodeArr[1]->parent=NULL;//每个树结点进行初始化
        for(int j=1;j<=n;++j){
            int left,right;
            scanf("%d%d",&left,&right);
            if(left!=-1){//左孩子不空
                nodeArr[j]->left=nodeArr[left];
                nodeArr[left]->parent=nodeArr[j];
            }else{
                nodeArr[j]->left=NULL;
            }
            if(right!=-1){//右孩子不空
                nodeArr[j]->right=nodeArr[right];
                nodeArr[right]->parent=nodeArr[j];
            }else{
                nodeArr[j]->right=NULL;
            }
        }
        //查询
        int lhs,rhs;
        for(int j=0;j<m;++j){
            scanf("%d%d",&lhs,&rhs);
            vector<int> lvec;//存放结点lhs到根的路径 
            TreeNode* p=nodeArr[lhs];
            while(p!=NULL){
                lvec.push_back(p->num);
                p=p->parent;
            }
            vector<int> rvec;//存放结点rhs到根的路径 
            p=nodeArr[rhs];
            while(p!=NULL){
                rvec.push_back(p->num);
                p=p->parent;
            }
            int l=lvec.size()-1;
            int r=rvec.size()-1;
            //是从根开始往下找的
            while(true){
                if(l<0||r<0||lvec[l]!=rvec[r]){//l<0,r<0子路径的情况
                    break;
                }
                --l;
                --r;
            }
            printf("%d\n",l+r+2);//公共结点前一个结点因此需要+2
        }
    }
}