【题目描述】
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
【思路】
考虑以下情况:
1.如果结点有右子树,那么它的下一个结点就是右子树的最左结点
2.如果结点没有右子树:
(1)如果结点是其父节点的左子树,那么它的下一个结点就是其父节点
(2)如果结点是其父节点的右子树,那么就向上一直遍历,直到它是父节点的左子节点的节点,那么这个节点的父节点就是我们需要的
【代码】
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if(pNode == NULL)
return NULL;
if(pNode->right)
{
TreeLinkNode* tmp = pNode->right;
while(tmp->left)
tmp = tmp->left;
return tmp;
}
else if(pNode->next)
{
TreeLinkNode* tmp = pNode;
TreeLinkNode* root = pNode->next;
while(root && tmp == root->right)
{
tmp = root;
root = root->next;
}
return root;
}
return NULL;
}