【牛客题解】 ——二叉树的下一个节点

二叉树的下一个节点

题中给出的指向父亲节点的指针，用next表示，有点会让人误解。

思路：我们以以下这棵二叉树为例，中序遍历为{D,B,H,E,I,A,F,C,G}

把很容易可以发现，如果要找一个节点的下一结点，比如说是B节点的下一结点，我们首先看B的右树是不是为空，如果不为空，一直沿着指向左子结点的指针找到的叶子节点即为下一个节点，在这里B的下一个节点是H；

无右子树，且结点是该结点父结点的左子树，则下一结点是该结点的父结点，例如 H，下一结点是 E

无右子树，且结点是该结点父结点的右子树，则我们一直沿着父结点追朔，直到找到某个结点是其父结点的左子树，如果存在这样的结点，那么这个结点的父结点就是我们要找的下一结点。例如 I，下一结点是 A；例如 G，并没有符合情况的结点，所以 G 没有下一结点

/*
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==NULL)
            return NULL;
        if(pNode->right!=NULL)  //右树不为空，右树的最左孩子就是他的下一结点
        {
            pNode=pNode->right;
            if(pNode->left!=NULL)
                pNode=pNode->left;
            return pNode;
        }
        while(pNode->next)  //右树为空
        {
            TreeLinkNode* parent=pNode->next;
            if(parent->left==pNode)
                return parent;
            pNode=pNode->next;
        }
        return NULL;
    }
};