剑指offer-二叉树的下一个节点(C++)

1、题目描述

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

2、解题思路

仔细观察，可以把中序下一结点归为几种类型：
1、 有右子树，下一结点是右子树中的最左结点，例如 B，下一结点是 H
2、 无右子树，且结点是该结点父结点的左子树，则下一结点是该结点的父结点，例如 H，下一结点是 E
3、无右子树，且结点是该结点父结点的右子树，则我们一直沿着父结点追朔，直到找到某个结点是其父结点的左子树，如果存在这样的结点，那么这个结点的父结点就是我们要找的下一结点。例如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 == nullptr)
            return pNode;
        if(pNode->right != nullptr){    //1：该节点有右子树
            TreeLinkNode* p = pNode->right;
            while(p->left != nullptr){
                p = p->left;
            }
            return p;
        }
        if(pNode->next != nullptr && pNode->next->left == pNode){    //2：该节点是其父节点的左子树
            return pNode->next;
        }
        if(pNode->next != nullptr && pNode->next->right == pNode){    //3：该节点是其父节点的右子树
            TreeLinkNode* p = pNode->next;
            while(p->next != nullptr && p == p->next->right){
                p = p->next;
            }
            return p->next;
        }
        return nullptr;
    }
};