Python实现二叉树的基础操作

本篇的主要内容:

  • Python实现二叉树的建立与遍历
  • 如何判断完全二叉树

最近遇到这个问题,要使用Python实现二叉树的一些操作,在网上这部分的资源不是很多,也没有找好很好的,只好按照自己的思路简单写了一下,本来算法是不局限于语言的,说是这么说,但是还是遇到了一些问题,在此梳理记录一下。

二叉树建立

首先定义二叉树的先序字符串,并使用#表示某个节点的子树为空的情况:

二叉树类的定义为:

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

创建二叉树:

pos = 0
def create(seq):
    global pos
    ch = seq[pos]
    pos += 1
    if ch == '#':
        return None
    else:
        temp = TreeNode(ch)
        temp.left = create(seq)
        temp.right = create(seq)
        return temp
 
seq = "abd###ce###"
root = create(seq)

这里创建的二叉树形状为:

         a
       /   \
      b     c
     /     /
  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
二叉树基本操作包括:创建二叉树,遍历二叉树,插入节点,删除节点等。下面是Python代码实现: ```python class Node: def __init__(self, val=None): self.val = val self.left = None self.right = None class BinaryTree: def __init__(self): self.root = None def insert(self, val): node = Node(val) if not self.root: self.root = node else: cur = self.root while True: if val < cur.val: if cur.left is None: cur.left = node break else: cur = cur.left else: if cur.right is None: cur.right = node break else: cur = cur.right def delete(self, val): def getMinNode(node): while node.left: node = node.left return node def _delete(node, val): if not node: return None if val < node.val: node.left = _delete(node.left, val) elif val > node.val: node.right = _delete(node.right, val) else: if not node.left and not node.right: node = None elif not node.left: node = node.right elif not node.right: node = node.left else: minNode = getMinNode(node.right) node.val = minNode.val node.right = _delete(node.right, minNode.val) return node self.root = _delete(self.root, val) def search(self, val): cur = self.root while cur: if val == cur.val: return True elif val < cur.val: cur = cur.left else: cur = cur.right return False def preorder(self, node): if node: print(node.val, end=' ') self.preorder(node.left) self.preorder(node.right) def inorder(self, node): if node: self.inorder(node.left) print(node.val, end=' ') self.inorder(node.right) def postorder(self, node): if node: self.postorder(node.left) self.postorder(node.right) print(node.val, end=' ') ``` 使用示例: ```python bt = BinaryTree() bt.insert(5) bt.insert(2) bt.insert(7) bt.insert(1) bt.insert(3) bt.insert(6) bt.insert(8) bt.preorder(bt.root) # 5 2 1 3 7 6 8 print() bt.inorder(bt.root) # 1 2 3 5 6 7 8 print() bt.postorder(bt.root) # 1 3 2 6 8 7 5 print() bt.delete(2) bt.inorder(bt.root) # 1 3 5 6 7 8 ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值