半✅\(^o^)/~
题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
1. 如果一节点的右子节点存在,在看右子节点有没有左子节点的递归。确定下一个节点。
2. 如果右子节点不存在,寻找父节点,如果他是父节点的右子树,则递归下去。
/*
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)
{
TreeLinkNode* pnext = nullptr;
if (pNode->right!=nullptr)
{
TreeLinkNode* pright= pNode->right;
while(pright->left!=nullptr)
{
pright=pright->left;
}
pnext= pright;
}
else if (pNode->next!=nullptr){
TreeLinkNode* pcurrent=pNode;
TreeLinkNode* pparent= pNode->next;
while(pparent!=nullptr&&pcurrent==pparent->right){
pcurrent=pparent;
pparent=pparent->next;
}
pnext=pparent;
}
return pnext;
}
};