找到二叉树中序遍历的下一个节点

给定一个二叉树其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示,从子节点指向父节点的用虚线表示


示例:
输入:{8,6,10,5,7,9,11},8
返回:9
解析:这个组装传入的子树根节点,其实就是整颗树,中序遍历{5,6,7,8,9,10,11},根节点8的下一个节点就是9,应该返回{9,10,11},后台只打印子树的下一个节点,所以只会打印9,如下图,其实都有指向左右孩子的指针,还有指向父节点的指针,下图没有画出来

数据范围:节点数满足  ,节点上的值满足 

要求:空间复杂度  ,时间复杂度 

class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None
        self.parent = None  # 父节点指针

def get_next(node):
    if node is None:
        return None

    # 情况1: 如果有右子树,则下一个节点是右子树的最左节点
    if node.right:
        node = node.right
        while node.left:
            node = node.left
        return node

    # 情况2和情况3: 如果没有右子树,则需要向上遍历父节点
    while node.parent:
        if node.parent.left == node:
            return node.parent
        node = node.parent

    # 情况4: 如果是最后一个节点,则没有下一个节点
    return None

# 构建示例树
# 为了简化,这里只构建了部分树结构
root = TreeNode(8)
root.left = TreeNode(6)
root.left.parent = root
root.right = TreeNode(10)
root.right.parent = root
root.left.left = TreeNode(5)
root.left.left.parent = root.left
root.left.right = TreeNode(7)
root.left.right.parent = root.left
root.right.left = TreeNode(9)
root.right.left.parent = root.right
root.right.right = TreeNode(11)
root.right.right.parent = root.right

# 测试
current_node = root.left  # 设置当前节点为节点6
next_node = get_next(current_node)
print(next_node.val if next_node else None)  # 应该输出7

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值