剑指offer.8-二叉树的下一个节点

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

我们就以上图中的二叉树为例,它的中序遍历是{4, 2, 8, 5, 9, 1, 6, 3, 7}。分析某个节点的下一个节点位置我们可以得到三种情况。

(设:输入节点为node,它的下一节点为nextNode)

情况一:如果node有右子节点,那么nextNode就是node的右子节点的最左子节点。

情况二:如果node没有右子节点,并且node是它父节点的右子节点,这种情况我们就沿着父节点向上遍历,直到遇到一个节点的左子节点是父节点的节点,这个节点就是nextNode。

情况三:如果node没有右子节点,并且node是它父节点的左子节点,这种情况最简单,nextNode就是它的父节点。

三种情况分析完后就很容易写出下面的代码了。

public void findNextNode(BinaryTreeNode node) {
    if (node != null) return;

    BinaryTreeNode nextNode = null;
    if (node.rightNode != null) { // node有右子节点
        BinaryTreeNode right = node.rightNode;
        // 循环找到右子节点的最左子节点
        while (right.leftNode != null) { 
            right = right.leftNode;
        }
        nextNode = right;
    } else if (node.parentNode != null) { // node没有右子节点但有父节点
        BinaryTreeNode current = node;
        BinaryTreeNode parent = node.parentNode;
        // 沿着父节点向上遍历,直到遇到一个节点的左子节点是父节点的节点
        while (parent != null && current.equals(parent.rightNode)) { 
            current = parent;
            parent = parent.parentNode;
        }
        nextNode = parent;
    }
    if (nextNode != null) {
        System.out.println("nextNode's value is " + nextNode.value);
    }
}

画图分析规律和可能性很重要!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值