二叉树的下一个节点
题目如下:
给定一颗二叉树和其中的一个节点,如何找出中序遍历序列的下一个节点?树中的节点除了有两个分别指向左、右节点的指针,还有一个指向父节点的指针。
我们可以先举个例子来具体化一下,太抽象了不好理解
下面是一颗二叉树
它的中序遍历序列为**{d,b,h,e,i,a,f,c,g}**
我们可以分为以下几个情况分析:
1、该节点有右子树(b节点)
由中序遍历的特性可以知道,它的下一个节点为右子树中的最左子节点,因此可以从右子节点出发一直沿着左子节点,就能找到下一节点。例如树中的b节点,它的下一个节点的寻找路径为e=h->null 因此下一节点为h
2、该节点没有右子树,且该节点是它父节点的左子节点(d节点)
这时候该节点的下一节点就应该是它的父节点。例如树中的d节点,它的下一节点就是d->father=b节点
3、该节点没有右子树,且该节点是它父节点的右节点(i节点)
这种情况比较复杂,我们需要沿着指向父节点的一直往上遍历,直到找到这样一个节点x:x是x父节点y的左子节点。那么y就是我们需要找的下一个节点。例如树中的i节点,查找下一节点的路径就是
i->e->b->a 此时a->LeftChild=b 所以a就是i的下一个节点
那要是找不到这样的节点呢?比如树中的g节点,查找路径为g->c->a 此时发现a没有了父节点,并且c为a的右子节点,此时g就没有下一个节点(null)
好了,所有的情况都分析清楚了,我们可以根据分类情况来写代码了
BinaryTreeNode* GetNext(BinaryTreeNode* pNode)
{
if(pNode==nullptr) return nullptr;
BinaryTreeNode* pNext=nullptr;
//分类1
if(pNode->m_pRight!=nullptr)
{
BinaryTreeNode* pRight=pNode->m_pRight;
while(pRight->m_pLeft!=nullptr)
pRight=pRight->m_pLeft;
pNext=pRight;
}
else if(pNode->m_Parent!=nullptr)
{
BinaryTreeNode* pCurrent=pNode;
BinaryTreeNode* pParent=pNode->m_Parent;
//分类2,3
while(pParent!=nullptr && pCurrent==pParent->m_pRight)
{
pCurrent=pParent;
pParent=pCurrent->m_pParent;
}
pNext=pParent;
}
return PNext;
}
——————————————————————————————————————————
参考书籍 《剑指offer 第二版》