//==================================================================
// 《剑指Offer——名企面试官精讲典型编程题》代码
// 作者：何海涛
//==================================================================
// 面试题8：二叉树的下一个结点
// 题目：给定一棵二叉树和其中的一个结点，如何找出中序遍历顺序的下一个结点？
// 树中的结点除了有两个分别指向左右子结点的指针以外，还有一个指向父结点的指针。
#include <iostream>
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
BinaryTreeNode* m_pParent;
};
// 找出中序遍历顺序的下一个结点
BinaryTreeNode* GetNext(BinaryTreeNode* pNode) {
// 空指针检查
if (pNode == nullptr) {
return nullptr;
}
BinaryTreeNode* pNext = nullptr;
if (pNode->m_pRight != nullptr) {
pNext = pNode->m_pRight;
while (pNext->m_pLeft != nullptr) {
pNext = pNext->m_pLeft;
}
} 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;
}
// 构建二叉树
BinaryTreeNode* CreateBinaryTreeNode(int value) {
BinaryTreeNode* pNode = new BinaryTreeNode();
pNode->m_nValue = value;
pNode->m_pLeft = nullptr;
pNode->m_pRight = nullptr;
pNode->m_pParent = nullptr;
return pNode;
}
分析过程:
