首页 > 代码库 > 寻找二叉树中的最低公共祖先结点----LCA(Lowest Common Ancestor )问题（递归）

寻找二叉树中的最低公共祖先结点----LCA(Lowest Common Ancestor )问题（递归）

2024-09-20 09:13:05   221人阅读

转自 剑指Offer之 - 树中两个结点的最低公共祖先

题目： 

求树中两个节点的最低公共祖先。

思路一： 

——如果是二叉树，而且是二叉搜索树，那么是可以找到公共节点的。

二叉搜索树都是排序过的，位于左子树的节点都比父节点小，而位于右子树上面的节点都比父节点大。

如果当前节点的值比两个结点 的值都大，那么最低的共同的父节点一定是在当前节点的左子树中，于是下一步遍历当前节点的左子节点。

如果当前节点的值比两个结点的值都小，那么最低的共同的父节点一定是在当前节点的右子树中，于是下一步遍历当前节点的右子节点。

这样从上到下，找到的第一个在两个输入结点的值之间的节点，就是最低的公共祖先。

题目和代码参考：http://blog.csdn.net/u012243115/article/details/46875537。

思路二：

    如果这棵树不是二叉搜索树，甚至连二叉树都不是，而只是普通的树。

——如果有指向父节点的指针，那么这个题目转换成了求，两个双向链表的第一个公共结点的问题。

思路三：

    这棵树是普通的树，而且这个树中的结点没有指向父节点的指针。

——遍历这个树，看以这个节点为根的子树是否包含这两个节点，如果包含，判断这个节点的子节点是否包含，

——知道子节点都不包含而这个当前的节点包含，那么这个节点就是最低的公共祖先。

ps.这里存在大量的重复遍历，效率不高。

思路三：

    这棵树是普通的树，而且这个树中的结点没有指向父节点的指针。

——用两个链表分别保存从根节点到输入的两个结点的路径，然后把问题转换成两个链表的最后公共节点。

代码：

（这里假设树是普通二叉树，用思路三求解）

#include <iostream>  
#include <list>  
using namespace std;  
  
struct TreeNode  
{  
    int m_nValue;  
    TreeNode *m_pLeft;  
    TreeNode *m_pRight;  
    TreeNode(){}  
    TreeNode(int value):m_nValue(value),m_pLeft(NULL),m_pRight(NULL){}  
};  
  
//得到pNode结点的路径，放入path中 也可以用栈来实现 ，递归的本质就是一个压栈和出栈的过程
// 本题可为剑指offer原题P252，其解法可参考面试题25(P143)
bool GetNodePath(TreeNode * pRoot , TreeNode *pNode , list<TreeNode *> &path)  
{  
    if(pRoot == NULL)  
        return false;  
    path.push_back(pRoot);  
      
    bool found = false;  
    if(pRoot == pNode)  
    {  
        found = true;  
        return found;  
    }  
    // 注意理解这里的递归问题
    found = GetNodePath(pRoot->m_pLeft , pNode , path) || GetNodePath(pRoot->m_pRight , pNode , path);      
    if(!found)  
        path.pop_back();  
    return found;  
}  
  
//找到两条路径的最后一个公共结点即是公共祖先  
TreeNode * GetLastCommonNode(const list<TreeNode *> &path1 , const list<TreeNode*> &path2)  
{  
    list<TreeNode *>::const_iterator iterator1 = path1.begin();  
    list<TreeNode *>::const_iterator iterator2 = path2.begin();  
    TreeNode *pLast = NULL;  
    while(iterator1 != path1.end() && iterator2 != path2.end())  
    {  
        if(*iterator1 == *iterator2)  
            pLast = *iterator1;  
        iterator1++;  
        iterator2++;          
    }  
    return pLast;  
}  
  
TreeNode * GetLastCommonNodeParent(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);  
}  
  
TreeNode * findLCA(TreeNode *pRoot , TreeNode *pNode1 , TreeNode *pNode2)  
{  
    if(pRoot == NULL)  
        return NULL;  
    if(pRoot == pNode1 || pRoot == pNode2)  
        return pRoot;  
    TreeNode *left_lca = findLCA(pRoot->m_pLeft , pNode1 , pNode2);  
    TreeNode *right_lca = findLCA(pRoot->m_pRight , pNode1 , pNode2);  
    if(left_lca && right_lca )  
        return pRoot;  
    return (left_lca != NULL) ? left_lca : right_lca;  
}  
  
int main()  
{  
    TreeNode *p1 = new TreeNode(1);  
    TreeNode *p2 = new TreeNode(2);  
    TreeNode *p3 = new TreeNode(3);  
    TreeNode *p4 = new TreeNode(4);  
    TreeNode *p5 = new TreeNode(5);  
    TreeNode *p6 = new TreeNode(6);  
    TreeNode *p7 = new TreeNode(7);  
    TreeNode *p8 = new TreeNode(8);  
    TreeNode *p9 = new TreeNode(9);  
    TreeNode *p10 = new TreeNode(10);  
    p1->m_pLeft = p2;  
    p1->m_pRight = p3;  
    p2->m_pLeft = p4;  
    p2->m_pRight = p5;  
    p3->m_pLeft = p6;  
    p3->m_pRight = p7;  
    p4->m_pLeft = p8;  
    p4->m_pRight = p9;  
    p5->m_pLeft = p10;  
    TreeNode *p;  
    p = GetLastCommonNodeParent(p1 , p8 , p7);//p8和p7的最近公共祖先  
    if(p)  
        cout<<"最近的公共祖先是p"<<p->m_nValue<<endl;  
    else  
        cout<<"不存在公共祖先"<<endl;  
    TreeNode *q;  
    q = findLCA(p1 , p8 , p7);//p8和p7的最近公共祖先  
    if(q)  
        cout<<"最近的公共祖先是p"<<q->m_nValue<<endl;  
    else  
        cout<<"不存在公共祖先"<<endl;  
}

findLCA参考：http://www.acmerblog.com/lca-lowest-common-ancestor-5574.html

 

寻找二叉树中的最低公共祖先结点----LCA(Lowest Common Ancestor )问题（递归）


联系
我们