找出中序遍历顺序的下一个结点
问题描述:
给定一颗二叉树和其中一个节点,找出中序遍历序列的下一个节点!树中的每个节点有 左/右/父 三个指针域。
提示:
- 不是所有的节点都有其下一个节点,所以结果可能为 NULL
- 根节点的父节点为 NULL
节点的定义如下:
struct TreeLinkNode
{
int val;
struct TreeLinkNode *left;
struct TreeLinkNode *right;
struct TreeLinkNode *next; //父节点的指针域
}
给定一个二叉树:
问题分析:
1. 如目标节点有 右节点,则根据 中序遍历的顺序 “左 中 右”,下一个节点就是目标节点的 右子树的最左节点。
2.1. 如目标节点无 右节点,则根据 中序遍历的顺序 “左 中 右”,下一个节点在目标节点的上方。如果目标节点为其父节点的左节点,则目标节点的父节点就是下一个节点。
2.2. 如目标节点无 右节点,则根据 中序遍历的顺序 “左 中 右”,下一个节点在目标节点的上方。如果目标节点为其父节点的右节点,则我们可以沿着父指针域一直向上遍历,直到找到一个是它父节点的左子节点,如果有这样的节点,那么这个节点的父节点就是我们要找的下一个节点。
代码实现如下:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if(pNode == NULL)
return NULL;
TreeLinkNode* pNext = NULL;
if(pNode->right != NULL) //目标节点的右节点存在
{
TreeLinkNode* pRight = pNode->right;
while(pRight->left != NULL) //循环找到最左节点
pRight = pRight->left;
pNext = pRight; //结果赋于pNext
}
else if(pNode->next != NULL)//目标节点的父节点存在(在分析中已知右节点不存在)
{
TreeLinkNode* pCurrent = pNode;
TreeLinkNode* pParent = pNode->next;
//如果该节点为父节点的右节点,则继续向上遍历
//遍历到 pParent == NULL 则到达了根节点
//遍历到 pParent->right != pCurrent 则找到目标节点,退出循环
while(pParent != NULL && pParent->right == pCurrent)
{
pCurrent = pParent;
pParent = pParent->next;
}
//如未进入上述循环则说明
//说明1. 此节点可能就是根节点
//说明2. 此节点可能是其父节点的左节点
pNext = pParent;
}
return pNext; //返回可能为NULL,也可能为有效节点
}