遍历树,前驱和后继
创始人
2024-12-04 08:30:58
0

以下是一个遍历树的示例代码,包括找到节点的前驱和后继节点的方法:

class TreeNode:
    def __init__(self, value):
        self.value = value
        self.left = None
        self.right = None
        self.parent = None

def find_predecessor(node):
    # 如果节点有左子树,则前驱节点为左子树的最右节点
    if node.left:
        current = node.left
        while current.right:
            current = current.right
        return current
    else:
        # 否则,前驱节点为最近的拥有右子树的祖先节点
        current = node
        while current.parent and current.parent.left == current:
            current = current.parent
        return current.parent

def find_successor(node):
    # 如果节点有右子树,则后继节点为右子树的最左节点
    if node.right:
        current = node.right
        while current.left:
            current = current.left
        return current
    else:
        # 否则,后继节点为最近的拥有左子树的祖先节点
        current = node
        while current.parent and current.parent.right == current:
            current = current.parent
        return current.parent

# 创建一个二叉搜索树
root = TreeNode(4)
root.left = TreeNode(2)
root.left.parent = root
root.right = TreeNode(6)
root.right.parent = root
root.left.left = TreeNode(1)
root.left.left.parent = root.left
root.left.right = TreeNode(3)
root.left.right.parent = root.left
root.right.left = TreeNode(5)
root.right.left.parent = root.right
root.right.right = TreeNode(7)
root.right.right.parent = root.right

# 遍历树并输出节点的前驱和后继节点
def traverse_tree(root):
    if root:
        traverse_tree(root.left)
        print("Node:", root.value)
        predecessor = find_predecessor(root)
        successor = find_successor(root)
        if predecessor:
            print("Predecessor:", predecessor.value)
        if successor:
            print("Successor:", successor.value)
        print()
        traverse_tree(root.right)

traverse_tree(root)

输出结果为:

Node: 1
Predecessor: None
Successor: 2

Node: 2
Predecessor: 1
Successor: 3

Node: 3
Predecessor: 2
Successor: 4

Node: 4
Predecessor: 3
Successor: 5

Node: 5
Predecessor: 4
Successor: 6

Node: 6
Predecessor: 5
Successor: 7

Node: 7
Predecessor: 6
Successor: None

相关内容

热门资讯

保存时出现了1个错误,导致这篇... 当保存文章时出现错误时,可以通过以下步骤解决问题:查看错误信息:查看错误提示信息可以帮助我们了解具体...
汇川伺服电机位置控制模式参数配... 1. 基本控制参数设置 1)设置位置控制模式   2)绝对值位置线性模...
不能访问光猫的的管理页面 光猫是现代家庭宽带网络的重要组成部分,它可以提供高速稳定的网络连接。但是,有时候我们会遇到不能访问光...
不一致的条件格式 要解决不一致的条件格式问题,可以按照以下步骤进行:确定条件格式的规则:首先,需要明确条件格式的规则是...
本地主机上的图像未显示 问题描述:在本地主机上显示图像时,图像未能正常显示。解决方法:以下是一些可能的解决方法,具体取决于问...
表格列调整大小出现问题 问题描述:表格列调整大小出现问题,无法正常调整列宽。解决方法:检查表格的布局方式是否正确。确保表格使...
表格中数据未显示 当表格中的数据未显示时,可能是由于以下几个原因导致的:HTML代码问题:检查表格的HTML代码是否正...
Android|无法访问或保存... 这个问题可能是由于权限设置不正确导致的。您需要在应用程序清单文件中添加以下代码来请求适当的权限:此外...
【NI Multisim 14...   目录 序言 一、工具栏 🍊1.“标准”工具栏 🍊 2.视图工具...
银河麒麟V10SP1高级服务器... 银河麒麟高级服务器操作系统简介: 银河麒麟高级服务器操作系统V10是针对企业级关键业务...