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

一. 题目

给定一颗二叉树和其中的一个节点，找出中序遍历序列的下一个节点。

注意:

(1)如果给定的节点是中序遍历序列的最后一个，则返回空节点;

(2)二叉树一定不为空，且给定的节点一定不是空节点;

示例1：
假定二叉树是：[2, 1, 3, null, null, null, null]， 给出的是值等于2的节点。
则应返回值等于3的节点。
解释：该二叉树的结构如下，2的后继节点是3。
  2
 / \
1   3

二: 思路

(1):如果node有右子节点，那么nextNode就是node的右子节点的最左子节点.
(2):如果node没有右子节点，并且node是它父节点的右子节点，这种情况我们就沿着父节点向上遍历,
    直到遇到一个节点的左子节点是父节点的节点，这个节点就是nextNode。
(3):如果node没有右子节点，并且node是它父节点的左子节点，这种情况最简单，nextNode就是它的父节点

三: 代码实现

public void findNextNode(BinaryTreeNode node) {
    if (node != null) return;
 
    BinaryTreeNode nextNode = null;
    if (node.rightNode != null) { // node有右子节点
        BinaryTreeNode right = node.rightNode;
        // 循环找到右子节点的最左子节点
        while (right.leftNode != null) { 
            right = right.leftNode;
        }
        nextNode = right;
    } else if (node.parentNode != null) { // node没有右子节点但有父节点
        BinaryTreeNode current = node;
        BinaryTreeNode parent = node.parentNode;
        // 沿着父节点向上遍历，直到遇到一个节点的左子节点是父节点的节点
        while (parent != null && current.equals(parent.rightNode)) { 
            current = parent;
            parent = parent.parentNode;
        }
        nextNode = parent;
    }
    if (nextNode != null) {
        System.out.println("nextNode's value is " + nextNode.value);
    }
}