剑指Offer - 二叉树的下一个节点

题目

给定一颗二叉树和其中一个节点，如何找出中序遍历序列的下一个节点？树中的节点除了两个分别指向左、右子节点的指针，还有一个指向父节点的指针。

分析

分情况讨论：
1、一个节点有右子树，那么它的下一个节点就是它的右子树中的最左子节点
2、一个节点没有右子树
（1）如果节点是它父节点的左子节点，那么它的下一个节点就是它的父节点
（2）如果节点是它父节点的右子节点，我们沿着指向父节点的指针一直向上遍历，直到找到一个是节点是它父节点的左子节点。如果这样的节点存在，那么这个节点的父节点就是我们要找的下一个节点。
如下图所示，5的下一个节点是1

代码

	// 参数： root表示二叉树的根节点 node表示给定的节点
    public TreeNode getNextNode(TreeNode root,TreeNode node) {
        // 二叉树为空
        if(root == null ) {
            return root;
        }
        // 二叉树只有一个节点
        if(root.right == null && root.left == null){
            return null;
        }
        TreeNode nextNode = null;
        if(node.right != null) {
            TreeNode rightNode = root.right;
            while(rightNode.left != null){
                rightNode = rightNode.left;
            }
            return rightNode;
        }else{
            // 父节点存在
            if(node.parent != null) {
                TreeNode temp = node;
                // 如果当前节点是父节点的左子节点，直接返回父节点
                if(node == node.parent.left){
                    return node.parent;
                }
                TreeNode parentNode = node.parent;
                // 如果当前节点是父节点的右子节点，则往上遍历,
                // 直到找到一个节点是它父节点的左子节点，然后返回该节点的父节点
                while(parentNode != null && parentNode.right == temp) {
                    temp = parentNode;
                    parentNode = parentNode.parent;
                }
                return parentNode;
            }
        }
        return nextNode;
    }