题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:
1)如果该结点为空,则返回空
2)如果该结点存在右结点,则从其右结点开始,沿着其右结点的左结点找,一直找到最左的结点,返回最左的结点;
3)如果该结点不存在右结点,则要找其父结点,然后判断该结点是不是其父结点的左结点,是的话,就返回其父结点;不是的话,再从该结点的父结点开始,找其父结点的父结点,判断其父结点是不是其父结点的父结点的左结点,是的话,返回其父结点的父结点;不是的话,再依此重复,直到找到,或者直到找到根节点还没有找到,就返回空。
/*
struct TreeLinkNode {
int val;
struct TreeLinkNode *left;
struct TreeLinkNode *right;
struct TreeLinkNode *next;
TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
}
};
*/
class Solution {
public:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if(!pNode)
return NULL;
if(pNode->right)
{
TreeLinkNode* pRight = pNode->right;
while(pRight->left)
pRight = pRight->left;
return pRight;
}
while(pNode->next)
{
TreeLinkNode* pRoot = pNode->next;
if(pRoot->left == pNode)
return pRoot;
pNode = pNode->next;
}
return NULL;
}
};