算法题16 二叉树最近公共父节点

 

找寻二叉树中两个节点的公共父节点中最近的那个节点
情况1. 节点只有left/right，没有parent指针，root已知

解析：对于此种情形，只需找到两个节点到根节点的路径，然后就相当于两个链表中找公共节点。

template<typename T> struct TreeNode1 { T data; TreeNode1* pLChild; TreeNode1* pRChild; }; #include <vector> // 找寻节点路径，倒序，根节点在最后 template<typename T> bool FindNodePath(TreeNode1<T>* pRoot, TreeNode1<T>* p, std::vector<TreeNode1<T>*>& path) { if(pRoot == NULL) return false; if(p == pRoot) { path.push_back(pRoot); return true } else if(FindNodePath(pRoot->pLChild, p, path)) { path.push_back(pRoot->pLChild); return true; } else if(FindNodePath(pRoot->pRChild, p, path)) { path.push_back(pRoot->pRChild); return true; } return false; } template<typename T> TreeNode1<T>* FindNearestParentNode(TreeNode1<T>* pRoot, TreeNode1<T>* p1, TreeNode1<T>* p2) { std::vector<TreeNode1<T>*> path1, path2; bool bFind = FindNodePath(pRoot, p1, path1); bFind &= FindNodePath(pRoot, p2, path2); if(!bFind) return NULL; TreeNode1<T>* pReturn = NULL; size_t minSize = path1.size() > path2.size() ? path2.size() : path1.size(); // 起始点设在可能出现共同节点的部分的起点 for(size_t i = path1.size() - minSize, j = path2.size()-minSize; i < path1.size() && j < path2.size(); ++i, ++j) { if(path1[i] == path2[j]) { pReturn = path1[i]; } } return pReturn; }

 

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

解析：比情况1要简单一些，只需找到两个节点的深度，即可当作两个链表求公共节点。

template<typename T> struct TreeNode2 { T data; TreeNode2* pLChild; TreeNode2* pRChild; TreeNode2* pParent; }; template<typename T> int GetNodeDeepth(TreeNode2<T>* pNode) { int nDeepth = 0; // 此处已判断节点为NULL的情况 while(pNode) { ++nDeepth; pNode = pNode->pParent; } return nDeepth; } template<typename T> TreeNode2<T>* FindNearestParentNode(TreeNode2<T>* p1, TreeNode2<T>* p2) { int nDeepth1 = GetNodeDeepth(p1); int nDeepth2 = GetNodeDeepth(p2); TreeNode2* pReturn = NULL; // 把深度大的放在1里 if(nDeepth1 < nDeepth2) { pReturn = p1; p1 = p2; p2 = pReturn; pReturn = NULL; } for(int i = 0; i < nDeepth1-nDeepth2; ++i) p1 = p1->pParent; while(p1) { if(p1 == p2) { pReturn = p1; break; } p1 = p1->pParent; p2 = p2->pParent; } return pReturn; }

 

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

解析：如果node节点的值在a，b之间，即 a=<node<= b （假设a < b），那么node就是所求节点。否则根据值的大小取其左孩子或右孩子继续判断。

 

template<typename T> TreeNode1<T>* FindNearestParentNode(TreeNode1<T>* pRoot, T a, T b) { if(a > b) { T temp = a; a = b; b = a; } while(pRoot) { if(pRoot->data >= a && pRoot->data <= b) return pRoot; else if(pRoot->data < a && pRoot->data < b) pRoot = pRoot->pLChild; else pRoot = pRoot->pRChild; } return NULL; }