剑指offer-----输出二叉树的后继节点(java版)

一 题目

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

二 例子

对于下面的二叉树，8的后继节点为9(题目说的中序遍历)，6的后继节点为7，5的后继节点为6

三 思路

思路很简单，

①如果节点有右子树，则该节点的后继节点就是往右子树出发，然后转到右子树的左子树，一直到左子树的左子树为空（即输入节点的右子树的最左子树，例如节点8，后继节点就是9）

②如果节点没有右子树，则向上寻找父节点，直到父节点的左子树等于当前节点，则该父节点就是后继节点（如节点7，没有右子树，则向上找，这时到6这个节点，因为6的左子树不等于7，所以继续向上找，这时到节点8，8的左子树等于当前节点6，所以返回8）

四 程序源码

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!=null&&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;
    }
}