剑指Offer——二叉树的下一个结点

题目描述：

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

解题思路：

首先，先复习下，二叉树的中序遍历：中序遍历首先遍历左子树，然后访问根结点，最后遍历右子树。如图所示二叉树：中序遍历的结果为：{D,B,H,E,I,A,F,C,G}

这道题需要分情况讨论：

1、 如果一个节点有右子树的时候，那么它的下一个节点就是它的右子树中的最左子节点。（比如说b的下一个节点是h）
2、 如果一个节点没有右子树的时候，并且节点是它的父节点的左子节点，那么该节点的下一个节点就是它的父节点。（比如说d的下一个节点是b）
3、 如果一个节点没有右子树，并且它还是它父节点的右子节点的话，那么我们可以看着上面这幅图，假定一个符合情况的节点是目标节点（比如说 i），我们可以沿着指向父节点的指针一直向上遍历，直到找到它的第一个右父节点。如果这样的节点存在的话，那么这个节点的父节点就是我们要找的下一个节点。

总结一下
就是，如果存在右子树，就从右子树开始，一直往左寻找。
如果不存在右子树，就判断是不是左子树，然后一直返回它的根节点就是我们要求的答案。

参考代码：

代码很巧妙

 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;
    }