【剑指offer】二叉树的下一个结点（二叉树）

题目描述

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

思路

分为两种情况
（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)
    {
        if (pNode == nullptr)
            return nullptr;

        // 如果有右子树：找右子树最左节点
        if (pNode->right!= nullptr){
            pNode = pNode->right;
            while (pNode->left!= nullptr)
                pNode = pNode->left;
            return pNode;
        }

        // 如果没有右子树；往上找父节点；得到某个是其父节点右子树的节点，返回该节点
        while(pNode->next!= nullptr){
            TreeLinkNode *pRoot = pNode->next;
            // 如果该节点是父节点的左孩子
            if(pRoot->left == pNode)
                return pRoot;
            pNode = pNode->next;
        }
        // 该节点为最后一个节点
        return nullptr;
    }
};