题目:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:
首先考虑各种情况,熟悉二叉树的中序遍历。
如果当前结点有右子树,那么找右子树的最左边的结点。
如果当前结点没有右子树,如果有父节点,如果该结点是父节点的左子树,那么下一个结点一定是父节点;如果该结点是父节点的右子树,那么下一个结点是该结点的父结点直到是其父节点的左子树为止,那么就找到了。如果直到根结点还是没有找到,那么就返回NULL。
class Solution {
public:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if(pNode == NULL)
return NULL;
if(pNode->right)//有右子树,直接找右子树中最左边的结点
{
pNode = pNode->right;
while(pNode->left)
pNode = pNode->left;
return pNode;
}
while(pNode->next!=NULL)//没有右子树,但是有父亲结点
{
if(pNode->next->left == pNode)//是父亲结点的左子树
return pNode->next;
pNode = pNode->next;
}
return NULL;
}
};