剑指Offer(第2版)——面试题8：二叉树的下一个节点

题目：

给定一棵二叉树和其中的一个节点，如何找出中序遍历序列的下一个节点？树中的节点除了有两个分别指向左，右子节点的指针，还有一个指向父节点的指针。

1.解题思路：

上图是有9个节点的二叉树，树中从父节点指向子节点的指针用实线表示，从子节点指向父节点的指针用虚线表示。它的中序遍历是d,b,h,e,i,a,f,c,g。

• 如果一个节点有右子树，那么它的下一个节点就是它的右子树的最左子节点。也就是说，从右子节点出发一直沿着指向左子节点的指针，我们就能找到它的下一个节点。例如，图中节点b的下一个节点是h，节点a的下一个节点是f。
• 如果一个节点没有右子树，并且这个节点是它父节点的左子节点，那么它的下一个节点就是它的父节点。例如，图中节点d的下一个节点是b，f的下一个节点是c。
• 如果一个节点既没有右子树，并且它还是父节点的右子节点，这种情形就比较复杂。我们可以沿着指向父节点的指针一直向上遍历，直到找到一个是自己父节点的左子节点的节点。如果这样的节点存在，那么这个节点的父节点就是我们要找的下一个节点。
  例如：为了找到节点 i 的下一个节点，我们沿着指向父节点的指针向上遍历，先到达节点e，由于节点e是它父节点的右节点，不满足条件，继续向上遍历到达节点b,节点b是它父节点a的左子节点，满足条件，因此节点b的父节点a就是节点 i 的下一个节点。
• 如果上面的方法都没有找到下一个节点，说明该节点不存在下一个节点，返回nullptr。
  例如：为了找到节点g的下一个节点，我们沿着指向父节点的指针向上遍历，先到达节点c。由于节点c是父节点a的右节点，不满足条件，我们继续向上遍历到达节点a。由于节点a是树的根节点。它没有父节点。因此节点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;
        }
        TreeLinkNode* pNext = NULL; //给定节点的下一个节点
        // 当前节点有右子树，那么它的下一个节点就是它的右子树中最左子节点
        if(pNode->right != NULL){
            TreeLinkNode* pRight = pNode->right;
            //循环一直找到最左节点
            while(pRight->left != NULL){
                pRight = pRight-> left;
            }
            pNext = pRight;
        }
        //当前节点无右子树，在有父节点的前提下，该节点不是左子节点就是右子节点。
        else if(pNode->next != NULL){
            // 当前节点
            TreeLinkNode* pCur = pNode;
            // 父节点
            TreeLinkNode* pPar = pNode->next;
            //当前节点还是父节点的右子节点时，则继续寻找父节点，直到跳出！
            //只要给定的条件为真，while 循环语句会重复执行一个目标语句
            while(pPar != NULL && pCur == pPar->right){
                pCur = pPar;
                pPar = pCur->next;
            }
            //是左节点的话直接返回父节点
            pNext = pPar;
        }
        return pNext;
    }
};