【两次过】二叉树的下一个结点

题目描述

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

---

解题思路：

分析二叉树的下一个节点，一共有以下情况：
1.二叉树为空，则返回空；
2.若节点右孩子存在，则设置一个指针从该节点的右孩子出发，一直沿着指向左子结点的指针找到的叶子节点即为下一个节点；
3.若节点右孩子不存在：

查看该节点是否是其父节点的左孩子，若不是则向上移动到其父节点，直到该节点是其父节点的左孩子，则返回父节点。

注意处理推到根节点也没有结果的情况，只需要加入node.next != null即可

/*
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){
            node = node.right;
            while(node.left != null)
                node = node.left;
            return node;
        }
        
        //没右子树，则找第一个当前节点是父节点左孩子的节点
        //如果退到根节点还没找到直接返回Node.next也没问题，因为此时node.next==null
        while(node.next != null && node.next.left != node)
            node = node.next;
        
        return node.next;
    }
}