57二叉树的下一个结点

题目描述

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



思路：
如果一个节点有右子树，那么他的下一个节点就是它的右子树中的最左节点。
如果一个节点没有右子树，如果节点是它父节点的左子节点，那么它的下一个节点就是它的父节点。
如果一个节点即没有右子树，并且它还是它父节点的右子节点，我们可以沿着指向父节点的指针一直向上遍历，直到找到一个是他父节点的左子节点的节点。

/*
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 node){
        if(node == null) return null;
        if(node.right!=null){ //1、右节点不为空
            //node 的下一个节点就是 当前node的右子树中，最左的节点
            node = node.right;
            while(node.left!=null) 
                node = node.left;
            return node;
        }
        else if(node.next!=null){//2右节点为空，分2种情况
            //2.1、node是父节点的左子节点，那么node的下一个节点就是它的父亲节点。
            if(node.next.left==node) return node.next;
            //2.2、node 即没有右子树，并且是父节点的右子节点,沿着指针一直往上遍历，
            //直到找到一个是他父节点的左子节点的节点。
            if(node.next.right==node){
                while(node.next!=null){
                    if(node.next.left==node) return node.next;
                    node = node.next;
                }
            }
        }
        return null;
    }
}