56. 二叉树的下一个结点《剑指Offer》(Java版)

题目描述

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

 

 

/*
public 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.right != null) {
            return getLeft(pNode.right);
        } else {
            return getFather(pNode);

        }
        // 如果没有右节点，就返回父节点
    }

    private TreeLinkNode getFather(TreeLinkNode pNode) {
        TreeLinkNode father = pNode.next;
        // 如果他是父亲节点的左节点
        if (father == null || father.left == pNode) {
            return father;
        } else {
            // 如果他是父亲节点的右节点，那么返回
            return getFather(father);
        }
    }

    private TreeLinkNode getLeft(TreeLinkNode right) {
        while (right.left != null) {
            right = right.left;
        }
        return right;
    }
}