二叉树的下一个节点

半✅\(^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;
    }
};