剑指 Offer-08 二叉树的下一个节点

299 篇文章 1 订阅
227 篇文章 1 订阅

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

此题在 LeetCode 中没有,我想是因为没法实现指向父节点的指针,但是我们可以写一个函数找到父亲节点,这都不重要,方法是最重要的:
当某个节点存在右孩子的时候,那该节点的下一个节点就是右孩子的最左边的孩子,类似于寻找二叉搜索树右子树的最小值,如图节点 2,该节点存在右子树,那么中序遍历下一个节点就是 4 的最左子树,也就是 8,是 2 的下一个节点

如果该节点不存在右孩子,那就看他是父亲节点的左孩子还是右孩子,如果是左孩子,那么下一个节点就是父亲节点,比如节点 1, 0 ,下一个节点都是他们的父亲节点,但是如果该节点是右孩子就比较复杂了,需要找到该节点父亲节点是左孩子的节点,举个例子,节点 7 是 4 的右孩子,那么看 4 是不是 4 的父亲节点的左孩子,不是,继续向上找,2 是 父亲节点 1 的左孩子,结束循环,说明 7 的下一个节点是 2 的父亲节点 3(这里有点小饶,一缕就清楚)
在这里插入图片描述

public TreeNode nextNode(TreeNode node) {
        TreeNode pNext = null;
        TreeNode parent = searchParent(root, node.getData());//找到该节点的父亲节点
        if (node.getRight() != null) {//如果存在右孩子,找右孩子的最左孩子就是下一个节点
            TreeNode temp = node.getRight();
            while (temp.getLeft() != null) {
                temp = temp.getLeft();
            }
            pNext = temp;
            // 如果不存在父亲节点,说明是根节点且没有右孩子,说明已经是最后一个节点了
        } else if (parent != null) {// 如果存在父亲节点,判断该节点是左孩子还有右孩子,左孩子直接返回父亲节点,右孩子需要单独判断
            TreeNode temp = node;
            TreeNode pParent = parent;
            if (pParent.getRight() != null && pParent.getRight().getData() == temp.getData()) {//是右孩子
                while (pParent != null && temp == pParent.getRight()) {
                //向上找到第一个是父亲节点是左孩子的情况
                    temp = pParent;
                    pParent = searchParent(root, pParent.getData());
                }
                pNext = pParent;
            }
            return pParent;
        }
        return pNext;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

_努力努力再努力_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值