二叉树的下一个节点

题目描述

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

• 思路
  首先需要知道中序遍历下各位置节点的下一个节点在哪。
  ①当给定的节点是父节点时，它的下一个节点是其右子树的最左节点(若无右子树则返回空)；
  ②当给定的节点是左节点时，它的下一个节点是其父节点；
  ③当给定的节点是右节点时，它的下一个节点是其最上的父节点。
  由于最后两种情况难以直接表达出来，可以转化为一种情况，即把所有的右节点往上递推，转化为二叉树根节点的左子树节点对待。最终无法这样转化的是其最右子树的右节点，会返回null(手画二叉树就很容易理解了)。代码如下：

//非父节点
        while(pNode.next != null){
            if(pNode.next.left == pNode){ //转化为左子树判断
                return pNode.next;
            }
            pNode = pNode.next; //向上递推
        }
        return null;

最终代码：

/*
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    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;
    }
}