使用python写一个二叉树

简介: 使用python写一个二叉树

可以使用Python的类来实现二叉树,每个节点包括一个值和指向左右子节点的引用。

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

class BinaryTree:
    def __init__(self, root):
        self.root = Node(root)

    def insert(self, value):
        self._insert(self.root, value)

    def _insert(self, node, value):
        if value < node.value:
            if node.left:
                self._insert(node.left, value)
            else:
                node.left = Node(value)
        else:
            if node.right:
                self._insert(node.right, value)
            else:
                node.right = Node(value)

    def search(self, value):
        return self._search(self.root, value)

    def _search(self, node, value):
        if node is None or node.value == value:
            return node

        if value < node.value:
            return self._search(node.left, value)

        return self._search(node.right, value)

使用示例:

# 创建一个二叉树
tree = BinaryTree(5)
tree.insert(3)
tree.insert(7)
tree.insert(1)
tree.insert(4)
tree.insert(6)
tree.insert(8)

# 搜索值为6的节点
result = tree.search(6)
if result:
    print("找到节点:", result.value)
else:
    print("未找到节点")

以上是一个简单的二叉树的实现,包括插入和搜索功能。你可以根据需要进行扩展,比如实现删除节点等操作。

相关文章
|
7月前
|
存储 算法 Python
python算法(三)——树、二叉树、AVL树
python算法(三)——树、二叉树、AVL树
38 0
|
7月前
|
算法 Python
Python算法——二叉树遍历
Python算法——二叉树遍历
93 0
|
8天前
|
存储 算法 Python
python常用算法(5)——树,二叉树与AVL树(一)
python常用算法(5)——树,二叉树与AVL树
|
1月前
|
C++ Python Java
Java每日一练(20230501) 路径交叉、环形链表、被围绕的区域
Java每日一练(20230501) 路径交叉、环形链表、被围绕的区域
46 0
Java每日一练(20230501) 路径交叉、环形链表、被围绕的区域
|
8天前
|
存储 算法 Shell
python常用算法(5)——树,二叉树与AVL树(三)
python常用算法(5)——树,二叉树与AVL树
|
8天前
|
算法 Python
python常用算法(5)——树,二叉树与AVL树(二)
python常用算法(5)——树,二叉树与AVL树
|
18天前
|
SQL 算法 数据可视化
LeetCode题目99:图解中叙遍历、Morris遍历实现恢复二叉树搜索树【python】
LeetCode题目99:图解中叙遍历、Morris遍历实现恢复二叉树搜索树【python】
|
18天前
|
存储 算法 数据可视化
python多种算法对比图解实现 验证二叉树搜索树【力扣98】
python多种算法对比图解实现 验证二叉树搜索树【力扣98】
|
1月前
|
Python 人工智能
讯飞星火、文心一言和通义千问同时编“贪吃蛇”游戏,谁会胜出?
讯飞星火、文心一言和通义千问同时编“贪吃蛇”游戏,谁会胜出?
62 1
讯飞星火、文心一言和通义千问同时编“贪吃蛇”游戏,谁会胜出?
|
1月前
|
算法 Java Go
Rust每日一练(Leetday0025) 矩阵置零、搜索二维矩阵、颜色分类
Rust每日一练(Leetday0025) 矩阵置零、搜索二维矩阵、颜色分类
43 0
Rust每日一练(Leetday0025) 矩阵置零、搜索二维矩阵、颜色分类