Python实现二叉树的定义、前序、中序、后序、层次遍历

定义二叉树类,实现二叉树的插入、查找和赋值:#!/usr/bin/python3# -*- coding: utf-8 -*-class BinaryTree: def __init__(self, root_obj): self.key = root_obj self.leftChild = None self.rightChild = None def insert_left(self, new_node):
摘要由CSDN通过智能技术生成

定义二叉树类,实现二叉树的插入、查找和赋值:

#!/usr/bin/python3
# -*- coding: utf-8 -*-


class BinaryTree:
    def __init__(self, root_obj):
        self.key = root_obj
        self.leftChild = None
        self.rightChild = None

    def insert_left(self, new_node):
        if self.leftChild is None:
            self.leftChild = BinaryTree(new_node)
        else:
            temp = BinaryTree(new_node)
            temp.leftChild = self.leftChild
            self.leftChild = temp

    def insert_right(self, new_node):
        if self.rightChild is None:
            self.rightChild = BinaryTree(new_node)
        else:
            temp = BinaryTree(new_node)
            temp.rightChild = self.rightChild
            self.rightChild = temp

    def get_right_node(self):
        return self.rightChild

    def get_left_node(self):
        return self.leftChild

    def set_root_val(self, obj):
        self.key = obj

    def get_root_val(self):
        return self.key


# tree = BinaryTree('root')
# print(tree.get_left_node())
# tree.insert_left('left001')
# tree.insert_right('right001')
# tree.get_left_node().insert_left('left002')
# print(tree.get_left_node().get_left_node().get_root_val())

实现二叉树的前序、中序、后序递归与非递归遍历,实现二叉树的层次遍历:

#!/usr/bin/pyth
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,我可以帮你解决这个问题。以下是使用二叉链表实现叉树的代码: ```python class TreeNode: def __init__(self, val): self.val = val self.left = None self.right = None class BinaryTree: def __init__(self): self.root = None def buildTree(self, node_list): """ 建立二叉树 :param node_list: 结点列表,按照层次,空结点用 None 表示 """ if not node_list: return None self.root = TreeNode(node_list[0]) node_queue = [self.root] front = 0 index = 1 while index < len(node_list): node = node_queue[front] front += 1 item = node_list[index] index += 1 if item is not None: left_node = TreeNode(item) node.left = left_node node_queue.append(left_node) if index >= len(node_list): break item = node_list[index] index += 1 if item is not None: right_node = TreeNode(item) node.right = right_node node_queue.append(right_node) def preorderTraversal(self, node): """ """ if node is None: return print(node.val, end=' ') self.preorderTraversal(node.left) self.preorderTraversal(node.right) def inorderTraversal(self, node): """ """ if node is None: return self.inorderTraversal(node.left) print(node.val, end=' ') self.inorderTraversal(node.right) def postorderTraversal(self, node): """ 后 """ if node is None: return self.postorderTraversal(node.left) self.postorderTraversal(node.right) print(node.val, end=' ') def levelTraversal(self, node): """ 按层 """ if node is None: return node_queue = [node] while len(node_queue) > 0: cur_node = node_queue.pop(0) print(cur_node.val, end=' ') if cur_node.left is not None: node_queue.append(cur_node.left) if cur_node.right is not None: node_queue.append(cur_node.right) def getDepth(self, node): """ 求树的深度 """ if node is None: return 0 left_depth = self.getDepth(node.left) right_depth = self.getDepth(node.right) return max(left_depth, right_depth) + 1 def getPath(self, node, target): """ 求指定结点到根的路径 """ path = [] self.__getPath(node, target, path) return path def __getPath(self, node, target, path): if node is None: return False if node.val == target: path.append(node.val) return True if self.__getPath(node.left, target, path) or self.__getPath(node.right, target, path): path.append(node.val) return True return False def destroyTree(self): """ 销毁二叉树 """ self.__destroyTree(self.root) def __destroyTree(self, node): if node is None: return self.__destroyTree(node.left) self.__destroyTree(node.right) del node def customOperation(self, node): """ 自定义操作 """ # 这里可以实现叉树的其他操作 pass if __name__ == '__main__': node_list = [1, 2, 3, None, None, 4, 5, None, None, None, None] bt = BinaryTree() bt.buildTree(node_list) print(":") bt.preorderTraversal(bt.root) print("\n:") bt.inorderTraversal(bt.root) print("\n后:") bt.postorderTraversal(bt.root) print("\n按层:") bt.levelTraversal(bt.root) print("\n树的深度:", bt.getDepth(bt.root)) path = bt.getPath(bt.root, 5) print("5号结点到根的路径:", path) bt.destroyTree() ``` 使用示例: ``` : 1 2 3 4 5 : 2 1 4 3 5 后: 2 4 5 3 1 按层: 1 2 3 4 5 树的深度: 3 5号结点到根的路径: [5, 2, 1] ``` 你可以在 `customOperation` 方法实现其他二叉树的操作。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值