面试题：二叉树的下一个节点

Posted 2021-01-01 aaron12

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

思路：

//包含指向父节点的指针
//node.next  node.left node.right
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;
        }
        //如果没有又子树 返回当前父节点是左子树的节点
        while(node.next!=null&&node!=node.next.left){
            node=node.next;
        }
        return node.next;
    }
}