给定一个二叉树其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的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