要求:二叉树中的下一个节点
给定一颗二叉树和其中的一个节点,如何找出中序遍历序列的下一个节点?树中的节点除了有两个分别指向左右节点的指针,还有一个指向父节点的指针。
struct BinaryTreeNode
{
int m_value;
BinaryTreeNode *m_pLeft;
BinaryTreeNode *m_pRight;
BinaryTreeNode *m_pParent;
};
测试用例:
- 普通二叉树(完全二叉树;不完全二叉树)
- 特殊二叉树(所有节点都没有右子节点的二叉树;所有节点都没有左子节点的二叉树;只有一个节点的二叉树;二叉树的根节点指针为nullptr)
- 不同位置的节点的下一个节点(下一个节点为当前节点的右子节点、右子树的最左子节点、父节点、跨层的父节点等;当前节点没有下一个节点)
本题考点:
- 考查对二叉树中序遍历的理解程度。只有对二叉树的遍历算法有了深刻的理解,应聘者才有可能准确找出每个节点的中序遍历的下一个节点。
- 考查分析复杂问题的能力。只有画出二叉树的结构图、通过具体的例子找出中序遍历下一个节点的规律,才有可能设计出可行的算法。
源代码:
BinaryTreeNode *GetNext(BinaryTreeNode *pNode)
{
if (nullptr == pNode)
return nullptr;
BinaryTreeNode *pNext = nullptr;
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_pParent != nullptr)
{
BinaryTreeNode *pCurrent = pNode;
BinaryTreeNode *pParent = pNode->m_pParent;
while (pParent != nullptr && pCurrent == pParent->m_pRight)
{
pCurrent = pParent;
pParent = pParent->m_pParent;
}
pNext = pParent;
}
return pNext;
}
参考代码:https://github.com/zhedahht/CodingInterviewChinese2/tree/master/08_NextNodeInBinaryTrees
自己代码:https://github.com/quinta2019/Offer/tree/master/08_Q_NextNodeInBinaryTrees