【剑指offer】面试题8-二叉树的下一个节点（图文描述清晰明了）

下面的next是节点的父节点信息，不要看漏了。（我第一次看了半天~~~）

struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next; // 这个next是指向父节点的，这里差点看漏了
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};

给一个二叉树的图举举栗子：

比如：

求F的下一个节点： 因为有右子树，所以直接找它的右子树中的最左节点H，这是H就是它的下一个节点。
求B的下一个节点： 该节点无右子树，所以需要找父节点，它的第一个父节点是D,切B是D 的左子树节点，所以D就是B的下一个节点。（同理： H的下一个节点就是E）
求D的下一个节点: 该节点无右子树，所以找父节点C，但是D是C的右子树节点，还要继续向上找父节点F, 这是C是F的左子树节点，所以D的下一个节点就是C的父节点F。等下代码中也是这样写的。
求G的下一个节点： 该节点无右子树，所以找父节点E，同样G是E的右子树节点，还需要向上找父节点F，但是E也是F的右子树节点，这是F就没有父节点了，所以G就没有下一个节点。

（中序遍历的顺序： A C B D F H E M G）

所以思路就很清晰了：

1. 如果该节点存在右子树，则它的下一个节点就是它的右子树中的最左节点（一定是该节点的右子树中）
2. 如果该节点不存在右子树，那它的节点必然是要找父节点。 (很关键的信息，根据中序遍历的概念得出)
3. 不存在右子树，而且的是父节点的左子树，那么它的下一个节点就是它本身的父节点。
4. 不存在右子树，同时是父节点的右子树，那么它的下一个节点需要一层一层往上找，直到找到一个节点是它的父节点的左子树节点，那么这个节点的父节点就要需要找的节点。

（ps: 这里代码可以直接将下面两种情况，一起讨论，直接判断这个节点是否为父节点的右子树，如果是，则一层一层往上找满足条件的节点，然后返回该节点的父节点信息；如果不是，也是返回该节点的父节点信息。）

牛客网AC代码如下

class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* p)
    {
        // 如果改节点存在右子树，则是右子树中的最左节点时该节点的下一个节点
        if(p->right) {
             p = p->right;
            while(p->left) p = p->left;
            return p;
        }
        //该节点无右子树，则需要找父节点
        // 1. 它在它的父节点的左子树中，则下一个节点就是父节点，
        // 2. 它在他的父节点的右子树中，则需要一直找到满足条件(一个节点是父节点的左节点)的父节点
        while(p->next && p == p->next->right) 
            p = p->next;
        return p->next;
    }
};