二叉树两个节点的公共节点

转自：http://blog.csdn.net/hhygcy/article/details/4660362

很流行的一个问题，常见于各种面试中，http://fayaa.com/tiku/view/16/ 这里有一个很好的汇总.

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

情况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)步之后，齐头并进，就能解决了。

[cpp]  view plain copy

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这里)。这样的问题，网上广为流传着类似的代码：

[cpp]  view plain copy

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,我们找到了我们的答案。下面的代码完全按照这个思路写成

[cpp]  view plain copy

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，然后才逐渐稳定下来，真的很难想象如果是在面试的环境下，在纸笔之上会有如何的表现，可能真是只有天知道了。