剑指Offer——8.二叉树的下一个节点

题目:

给定一颗二叉树和其中一个节点,如何找出中序遍历序列的下一个节点?树中的节点除了有两个分别指向左、右子节点的指针,还有一个指向父节点的指针。二叉树节点定义如下:

    class BinaryTreeNode {
        int data;
        BinaryTreeNode left;
        BinaryTreeNode right;
        BinaryTreeNode parent;
	
        BinaryTreeNode(int data) {
		    this.data = data;
	    }
    }

算法思想:

1.如果该节点有右子树,那么它的下一个节点就是它的右子树的左子节点。

2.如果该节点没有右子树,并且是它的父节点的左子节点,那么它的下一个节点就是它的父节点。

3..如果该节点没有右子树,并且是它的父节点的右子节点,那么它的下一个节点就是它的第一个有左子节点的祖先节点(不包含父节点);否则没有下一个节点。

代码实现:

    public BinaryTreeNode getNext(BinaryTreeNode pNode) {
        if(pNode == null) 
            return null;
        
        if(pNode.right != null) {    //节点有右子树
            BinaryTreeNode node = pNode.right;    //得到右子节点
            while(node.left != null)     
                node = node.left;
            return node;
        } else {    //节点没有右子树
            while (pNode.parent != null) {
                BinaryTreeNode node = pNode.parent;    //得到父节点
                if (node.left == pNode)    //该节点是父节点的左子节点
                    return node;
                pNode = pNode.parent;    //得到父节点的父节点
            }
        }
        
        return null;
    }

猜你喜欢

转载自blog.csdn.net/BlackMaBa/article/details/81459445