剑指57 二叉树的下一个结点

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

分析：

• 首先我们需要清楚二叉树中序遍历的规则是： 左 —> 根 —> 右
  

• 对于该树，将其中序遍历后的结果为：DBHEIAFCG
  根据中序遍历的规则，可以将中序下一结点分为以下3种情况：

• 1当前结点有右子树，下一结点是当前节结点右子树中的最左结点，例如对B结点而言，其下一个结点是H

• 2当前结点无右子树，且结点是该结点父结点的左子树，则下一结点是该结点的父结点，例如对结点H而言，下一结点为E

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

• 代码

public TreeLinkNode GetNext(TreeLinkNode pNode){
        if(pNode == null) return null;
        
        //如果当前结点的右子树是不为空
        if(pNode.right != null){
            pNode = pNode.right;
            while(pNode.left != null)
                pNode = pNode.left;
            return pNode;
        }
        //当前结点无右子树时，判断结点是父结点的左子树还是右子树
        while(pNode.next != null){
            if(pNode.next.left == pNode)
                return pNode.next;
            pNode = pNode.next;
        }
        return null;
    }

public class Solution {

public TreeLinkNode GetNext(TreeLinkNode pNode) {
     // 1
     if (pNode.right != null) {
         TreeLinkNode pRight = pNode.right;
         while (pRight.left != null) {
             pRight = pRight.left;
         }
         return pRight;
     }
     // 2
     if (pNode.next != null && pNode.next.left == pNode) {
         return pNode.next;
     }
     // 3
     if (pNode.next != null) {
         TreeLinkNode pNext = pNode.next;  //pNext表示父结点
         while (pNext.next != null && pNext.next.right == pNext) {
             pNext = pNext.next;
         }
         return pNext.next;
     }
     return null;
 }
}