`
wusuoya
  • 浏览: 629441 次
  • 性别: Icon_minigender_2
  • 来自: 成都
社区版块
存档分类
最新评论

找寻二叉树中两个节点的公共父节点中最近的那个节点

阅读更多

情况1. 节点只有left/right,没有parent指针,root已知

情况2. root未知,但是每个节点都有parent指针

情况3. 二叉树是个二叉查找树,且root和两个节点的值(a, b)已知


虽然情况一是第一个情况,但是看上去比较复杂,我们放到最后来说,先从第二个情况开始说。

                                             10

                                          /       \
                                        6         14
                                      /  \       /   \
                                    4   8   12   16

                                   /  \

                                  3   5

画一个二叉树来做例子。如果我们要找3和8这两个节点的公共父亲节点,我们的做法是首先找到3到根节点的路劲,然后找到8到根节点的路径。

                                             10

                                          //       \
                                        6         14
                                              /   \
                                    4   8   12   16

                                   /   \

                                  3   5

3的路径用红色表示,8的用绿色表示,可以看到, 这里的问题实际上是另一个我们熟知的问题,有2个相交的单链表,找出它们的相交点!

只要把这个二叉树的图片倒过来看,或者把脖子倒过来看就知道了:)那个方法也是传统的求出linkedList A的长度lengthA, linkedList B的长度LengthB。然后让长的那个链表走过abs(lengthA-lengthB)步之后,齐头并进,就能解决了。

  1. int  getLength (bstNode* pNode)  
  2. {     
  3.     int  length = 0;  
  4.     bstNode* pTemp = pNode;  
  5.     while  (pTemp)  
  6.     {  
  7.         length ++ ;  
  8.         pTemp = pTemp->pParent;  
  9.     }  
  10.     return  length;  
  11. }  
  12. bstNode* findLCACase2(bstNode* pNode1, bstNode* pNode2)  
  13. {  
  14.     int  length1 = getLength(pNode1);  
  15.     int  length2 = getLength(pNode2);  
  16.       
  17.     // skip the abs(length1-length2)   
  18.     bstNode* pIter1 = NULL;  
  19.     bstNode* pIter2 = NULL;  
  20.     int  k=0;  
  21.     if  (length1>=length2)  
  22.     {  
  23.         bstNode* pTemp = pNode1;  
  24.         while  (k++<length1-length2)  
  25.         {  
  26.             pTemp = pTemp->pParent;   
  27.         }  
  28.         pIter1 = pTemp;  
  29.         pIter2 = pNode2;  
  30.     }  
  31.     else   
  32.     {  
  33.         bstNode* pTemp = pNode1;  
  34.         while  (k++<length2-length1)  
  35.         {  
  36.             pTemp = pTemp->pParent;   
  37.         }  
  38.         pIter1 = pNode1;  
  39.         pIter2 = pTemp;  
  40.     }  
  41.       
  42.     while  (pIter1&&pIter2&&pIter1!= pIter2)  
  43.     {  
  44.         pIter1 = pIter1->pParent;  
  45.         pIter2 = pIter2->pParent;  
  46.     }  
  47.     return  pIter1;  
  48. }  

自己写了个代码,总觉得有些拖沓冗余,希望有缘人看到文章之后能帮我改写的更和谐一些。

还是原来这个图,情况三,如果是个二叉搜索树,而且root和a, b已知,我们这个case假设a,b=3,8。从知道根这个条件我们很自然联想到递归(当然不递归也可以)地往下找。关键是收敛条件,什么情况下可以判断 出当然检查的这个节点是最近父亲节点呢?其实从这个例子已经可以看出一些端倪了,如果当前访问的节点比a,b来的都小,肯定不行。如果比a,b都大,也不 行。那也就是说,这个节点只有在a<=node<=b的区间内才成立(我们假定a<b这里)。这样的问题,网上广为流传着类似的代码:

  1. bstNode* findLCACase3(bstNode* pNode,  int  value1,  int  value2)  
  2. {  
  3.     bstNode* pTemp = pNode;  
  4.     while  (pTemp)  
  5.     {  
  6.         if  (pTemp->data>value1 && pTemp->data>value2)  
  7.             pTemp = pTemp->pLeft;  
  8.         else   if (pTemp->data<value1 && pTemp->data<value2)  
  9.             pTemp = pTemp->pRight;  
  10.         else   
  11.             return  pTemp;  
  12.     }  
  13.     return  NULL;  
  14. }  

好,前面的问题都解决了,我们再回过头来看第一个情况,只有ROOT和left, right节点,没有parent也不是排序树,怎么办?网络上也流传着很多所谓的LCA,RMQ算法,我们不暇找个最合适的,尤其是在面试的时候,特定 时间空间下你很难写出一个逻辑非常复杂的东西(比如你会在面试的时候去实现一个Suffix Tree还是用动态规划来求最长公共子串,哪怕效率不同,我也选择动态规划:))。所以这里,碰到类似的问题的时候,我选择简单的记录找到node1和 node2的路径,然后再把它们的路径用类似的情况二来做分析,比如还是node1=3,node2=8这个case.我们肯定可以从根节点开始找到3这 个节点,同时记录下路径3,4,6,10,类似的我们也可以找到8,6,10。我们把这样的信息存储到两个vector里面,把长的vector开始的多 余节点3扔掉,从相同剩余长度开始比较,4!=8, 6==6, coooool,我们找到了我们的答案。下面的代码完全按照这个思路写成

  1. #include <vector>   
  2. bool  nodePath (bstNode* pRoot,  int  value, std::vector<bstNode*>& path)  
  3. {  
  4.     if  (pRoot==NULL)  return   false ;  
  5.     if  (pRoot->data!=value)  
  6.     {  
  7.         if  (nodePath(pRoot->pLeft,value,path))  
  8.         {  
  9.             path.push_back(pRoot);  
  10.             return   true ;  
  11.         }  
  12.         else   
  13.         {  
  14.             if  (nodePath(pRoot->pRight,value,path))  
  15.             {  
  16.                 path.push_back(pRoot);  
  17.                 return   true ;  
  18.             }  
  19.             else   
  20.                 return   false ;  
  21.         }  
  22.     }  
  23.     else   
  24.     {  
  25.         path.push_back(pRoot);  
  26.         return   true ;  
  27.     }  
  28. }  
  29. bstNode* findLCACase1(bstNode* pNode, int  value1,  int  value2)  
  30. {  
  31.     std::vector<bstNode*> path1;  
  32.     std::vector<bstNode*> path2;  
  33.     bool  find =  false ;  
  34.     find |= nodePath(pNode, value1, path1);  
  35.     find &= nodePath(pNode, value2, path2);  
  36.     bstNode* pReturn=NULL;  
  37.     if  (find)  
  38.     {  
  39.         int  minSize = path1.size()>path2.size()?path2.size():path1.size();  
  40.         int  it1 = path1.size()-minSize;  
  41.         int  it2 = path2.size()-minSize;  
  42.         for  (;it1<path1.size(),it2<path2.size();it1++,it2++)  
  43.         {  
  44.             if  (path1[it1]==path2[it2])  
  45.             {  
  46.                 pReturn = path1[it1];  
  47.                 break ;  
  48.             }  
  49.         }  
  50.     }  
  51.     return  pReturn;  
  52. }  

这段代码经历了大概30分钟的修改和debug,然后才逐渐稳定下来,真的很难想象如果是在面试的环境下,在纸笔之上会有如何的表现,可能真是只有天知道了。

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics