树中的两个结点之间的距离

/*
1) node1是node2的祖先节点或孩子节点，
可以理解为两个节点在一条线上。 例如：Dist(2,4), Dist(6,1)
2) node1 和 node2 没有直接或间接的父子关系。
例如，Dist(4,3)， 他们需要一个共同的祖先节点1 连接起来
Dist(n1, n2) = Dist(root, n1) + Dist(root, n2) - 2*Dist(root, lca)


*/

#include<iostream>
using namespace std;
struct Node{
    int data;
    Node *lchild;
    Node *rchild;
};
int max(int a,int b)
{
    return a>b?a:b;
}
Node *getPublicParetnNode(Node *root,int n1,int n2)
{
    if(root==NULL) return NULL;
    
    if(root->data==n1||root->data==n2)
    return root;
    
    Node *lNode=getPublicParetnNode(root->lchild,n1,n2);
    Node *rNode=getPublicParetnNode(root->rchild,n1,n2);
    
    if(!lNode&&!rNode) return NULL;
    else if(lNode&&rNode) return root;
    else return lNode==NULL?rNode:lNode;
}

int getDistanceToRoot(Node *root,int n)
{
    if(root==NULL)return -1;
    if(root->data==n) return 0;
    
    int level=getDistanceToRoot(root->lchild,n);
    
    if(level==-1)//表示左子树上没有。
    level=getDistanceToRoot(root->rchild,n);
    if(level!=-1)
      return level+1;
    return -1;
}

void buildTree(Node *&root)
{
    int num;
    scanf("%d",&num);
    if(num==0)
    {
        root=NULL;
        return;
    }    
    root=new Node;
    root->data=num;
    buildTree(root->lchild);
    buildTree(root->rchild);
}

int getTwoNodeDis(Node *root,int n1,int n2)
{
    int d1=getDistanceToRoot(root,n1);
    int d2=getDistanceToRoot(root,n2);
    Node *node=getPublicParetnNode(root,n1,n2);
    int d3=getDistanceToRoot(root,node);
    return d1+d2-2*d3;
}