二叉树中序遍历的下一个结点

题目

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

思路

来分析一下，中序遍历，先访问左子树，然后是根，最后是右子树；也就是说，当前结点有没有左子树对该问题没有影响，因为，中序遍历的下一个结点一定不可能在左子树中，所以我们主要关注当前结点有没有右子树。

1）当前结点有右子树，那么下一个结点一定在右子树中， 且右子树中最左下的那个结点就是答案；
2）当前结点没有右子树，这个时候就要回溯了，向父节点回溯，如果发现当前结点是它父亲的左孩子则直接返回其父节点，如果是它父亲的右孩子则说明父结点已经被访问过了，继续回溯，直到当前结点是它父亲的左孩子为止；如果回溯到根结点还没有结束，那么说明当前结点是整个二叉树最右下的那个结点，也就是说它没有下一个结点了。

注意：

  因为要回溯到父节点，所以树中的结点不仅包含左右子结点left和right，同时包含指向父结点的指针next！

package com.zhumq.leetcode;
public class NextNodeOfInorder {
    public class TreeLinkNode {
        int val;
        TreeLinkNode left = null;
        TreeLinkNode right = null;
        //指向父节点的指针
        TreeLinkNode next = null;

        TreeLinkNode(int val) {
            this.val = val;
        }
    }
    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;
    }
}