题目:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针
思路:
若存在右孩子,直接在右孩子中寻找,若不存在右孩子,向上寻找第一个为左孩子的父节点
代码:
class Solution {
public:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if(NULL != pNode->right)
{
TreeLinkNode* Tmp = pNode->right;
while(Tmp->left != NULL)
{
Tmp = Tmp->left;
}
return Tmp;
}
TreeLinkNode* tmp = pNode;
while(NULL != tmp)
{
if(NULL != tmp->next && tmp == tmp->next->left)
{
return tmp->next;
}
tmp = tmp->next;
}
return NULL;
}
};