题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
/*
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 == nullptr) {
return nullptr;
}
TreeLinkNode* pNext = nullptr;
//存在右子树的情况
if(pNode -> right != nullptr) {
TreeLinkNode* pRight = pNode -> right;
while(pRight -> left != nullptr) {
pRight = pRight -> left;
}
pNext = pRight;
}
//无右子树的情况, else千万不能少,必须先判断无右子树
else if(pNode -> next != nullptr) {
TreeLinkNode* pCurrent = pNode;
TreeLinkNode* pParent = pNode -> next;
while(pParent != nullptr && pParent -> left != pCurrent) {
pCurrent = pParent;
pParent = pParent -> next;
}
pNext = pParent;
}
return pNext;
}
};