(剑指Offer)面试题50:树中两个结点的最低公共祖先
题目:
求树中两个结点的最低公共祖先
思路:
考虑一下几种情况:
1、该树为二叉搜索树
二叉搜索树是排序树,位于左子树点的结点都比父结点小,而位于右子树的结点都比父结点大,只需要从树的根结点开始和两个输入的结点进行比较。
如果当前结点的值比两个结点的值都大,那么最低的公共父结点一定在左子树,下一步就是遍历左子树;
如果当前结点的值比两个结点的值都小,那么最低的公共父结点一定在右子树;下一步就是遍历右子树;
如果当前结点的值介于两个结点的值之间,那么它就是两个结点的公共父结点,第一个找到的就是最低的公共父结点。
2、该树为二叉树,结点中有指向父结点的指针
有了父结点,就可以找到任意结点到根结点的路径;因此:
分别找到从两个结点开始到根结点的路径,即两个链表;
然后找到两个链表的第一个公共结点,就是最低的公共父结点;
3、该树为普通的树
从根结点开始,通过广度优先搜索,分别找到到达两个结点的路径;
然后找到两条路径的第一个公共结点,就是最低的公共父结点;
代码:
暂时先实现第三种条件
struct TreeNode{ int val; vector<TreeNode*> children; }; bool GetNodePath(TreeNode* pRoot,TreeNode* pNode,list<TreeNode*> &path){ if(pRoot==pNode) return true; path.push_back(pRoot); vector<TreeNode*>::iterator it=pRoot->children.begin(); bool found = false; while(!found && it!=pRoot->children.end()){ found=GetNodePath(*it,pNode,path); ++it; } if(!found) path.pop_back(); return found; } TreeNode* GetLastCommonNode(const list<TreeNode*> &path1,const list<TreeNode*> &path2){ list<TreeNode*>::const_iterator it1=path1.begin(); list<TreeNode*>::const_iterator it2=path2.begin(); TreeNode* pLast=NULL; while(it1!=path1.end() && it2!=path2.end()){ if(*it1==*it2) pLast=*it1; it1++; it2++; } return pLast; } TreeNode* GetLastCommonParent(TreeNode* pRoot,TreeNode* pNode1,TreeNode* pNode2){ if(pRoot==NULL || pNode1==NULL || pNode2==NULL) return NULL; list<TreeNode*> path1; GetNodePath(pRoot,pNode1,path1); list<TreeNode*> path2; GetNodePath(pRoot,pNode2,path2); return GetLastCommonNode(path1,path2); }