三叉链表存储的二叉树中查找最短路径
你可以使用广度优先搜索 (BFS) 算法来找到从 PA 到 PB 的最短路径,并打印路径上所有节点的值。下面是一个示例的 Python 代码:
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
self.parent = None
def find_shortest_path(root, PA, PB):
# 首先使用BFS找到从PA到PB的最短路径
queue = [(PA, [PA.val])]
while queue:
node, path = queue.pop(0)
if node == PB:
return path
if node.left:
queue.append((node.left, path + [node.left.val]))
if node.right:
queue.append((node.right, path + [node.right.val]))
if node.parent:
queue.append((node.parent, path + [node.parent.val]))
return []
# 使用示例
# 创建一个三叉链表存储的二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.parent = root
root.right.parent = root
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.left.left.parent = root.left
root.left.right.parent = root.left
root.left.right.left = TreeNode(6)
root.left.right.left.parent = root.left.right
# 定义PA和PB
PA = root.left.left # 节点4
PB = root.left.right.left # 节点6
# 打印从PA到PB的最短路径上所有节点的值
shortest_path = find_shortest_path(root, PA, PB)
for node_val in shortest_path:
print(node_val)
以上代码将打印从节点 4 到节点 6 的最短路径上所有节点的值。你可以根据你的实际树结构和节点定义进行适当的修改和调整。希望对你有帮助!如果你有任何问题,请随时问我。
原文地址: https://www.cveoy.top/t/topic/bG7x 著作权归作者所有。请勿转载和采集!