输入两个树节点,求它们的最低公共祖先。
1.如果是二叉搜索树,二叉搜索树是排序过的,位于左子树的节点都比父节点小,位于右子树的节点都比父节点大。
我们只需要从树的根节点开始和两个输入的节点进行比较。如果当前节点的值比两个节点的值都大,最低公共祖先一定在当前节点的左子树中,下一步比那里当前节点的左子树。反之则反之。
2.如果不是二叉搜索树,但是树的节点中有指向父节点的指针。这个问题可以转换为求两个链表的第一个公共节点。
从树的每一个叶子节点开始都有一个由指向父节点的指针串起来的链表,这些链表的尾指针都是树的根节点。
输入两个节点,那么这两个节点位于两个链表上,他们的最低公共祖先刚好就是这两个链表的第一个公共节点。
比如输入F和H,F在链表F-D-B-A上,而H在链表H-E-B-A上,这两个链表的第一个交点B刚好也是它们的最低公共祖先。
3.不是二叉搜索树,树的节点中也没有指向父节点的指针。
用两个链表分别保存从根节点到输入的两个节点的路径,然后把问题转换成两个链表的最后公共节点。
需要遍历两次树得到两个链表,每次时间复杂度O(n),
得到的两条路径的长度在最差情况下是O(n),通常情况下两条路径的长度是O(logn)。
整个程序的时间复杂度3O(n),空间复杂度是2O(n)。
#include <iostream>
#include <vector>
#include <list>
using namespace std;
typedef struct TreeNode
{
int data;
struct TreeNode *lchild;
struct TreeNode *rchild;
}TreeNode,*BiTree;
//按先序序列创建二叉树
int CreateBiTree(BiTree &T)
{
int data;
//按先序次序输入二叉树中结点的值(一个字符),‘#’表示空树
cin>>data;
if(data == -1) //叶子节点的左右孩子都为-1
{
T = NULL;
}
else
{
T = (BiTree)malloc(sizeof(TreeNode));
T->data = data;//生成根结点
CreateBiTree(T->lchild);//构造左子树
CreateBiTree(T->rchild);//构造右子树
}
return 0;
}
//得到从根节点pRoot开始到达节点pNode的路径,保存在path中
bool GetNodePath(BiTree pRoot,int val, list<BiTree> &path)
{
if (pRoot==NULL)
return false;
path.push_back(pRoot);
if (pRoot->data==val)
return true;
else
{
bool found;
found=GetNodePath(pRoot->lchild,val,path);
if(!found)
found=GetNodePath(pRoot->rchild,val,path);
if (!found)
path.pop_back();
return found;
}
}
//得到两个path的最后一个公共节点
BiTree GetLastCommonNode(const list<BiTree>& path1, const list<BiTree>& path2)
{
list<BiTree>::const_iterator i1=path1.begin();
list<BiTree>::const_iterator i2=path2.begin();
BiTree pLast=NULL;
while(i1!=path1.end()&&i2!=path2.end())
{
if (*i1==*i2)
{
pLast=*i1;
}
i1++;
i2++;
}
return pLast;
}
BiTree GetLastCommonParent(BiTree pRoot,int val1,int val2)
{
if (pRoot==NULL)
return NULL;
list<BiTree> path1;
GetNodePath(pRoot,val1,path1);//O(n)
list<BiTree> path2;
GetNodePath(pRoot,val2,path2);//O(n)
return GetLastCommonNode(path1,path2);//O(n)
}
void main()
{
BiTree T,result;
CreateBiTree(T);
int val1,val2;
while(cin>>val1>>val2)
{
result=GetLastCommonParent(T,val1,val2);
cout<<result->data<<endl;
}
}
原文:https://blog.csdn.net/wtyvhreal/article/details/45565383