python 二叉树实现

Posted 7749ha

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python 二叉树实现相关的知识,希望对你有一定的参考价值。

二叉树实现思想

  1、把每个节点都看作是一个对象包含以下特征:

      节点的当前值

      节点的左孩子(存储比当前节点值小的节点对象)

      节点右孩子(存储比当前节点值大的节点对象)

  2、二叉树就是以根节点开始的连续的对象串

代码实现:

# -*- coding:utf-8  -*-
# 日期:2018/6/12 8:09
# Author:小鼠标


# 节点对象
class Node:
    def __init__(self):
        self.left_children = None
        self.right_children = None
        self.value = None

# 二叉树对象
class tree:
    def __init__(self):
        self.root = False
        self.front_list = []
        self.middle_list = []
        self.after_list = []
    # 生成二叉树
    def create_tree(self,n=0,l=[]):
        if l == []:
            print("传入的列表为空")
            return
        if n > len(l)-1:
            print("二叉树生成")
            return
        node = Node()
        node.value = l[n]
        if not self.root:
           self.root = node
           self.list = l
        else:
            self.add(self.root,node)
        self.create_tree(n+1,l)
    # 添加节点
    def add(self,parent,new_node):
        if new_node.value > parent.value:
            # 插入值比父亲值大,所以在父节点右边
            if parent.right_children == None:
                parent.right_children = new_node
            else:
                self.add(parent.right_children,new_node)
        else:
            # 插入值比父亲值小,所以在父节点左边
            if parent.left_children == None:
                parent.left_children = new_node
            else:
                self.add(parent.left_children,new_node)
    # 前序(先中再左最后右)
    def front(self,node=None):
        if node == None:
            node = self.root
        # 输出当前节点
        self.front_list.append(node.value)
        # 先判断左枝
        if not node.left_children == None:
            self.front(node.left_children)
        # 再判断右枝
        if not node.right_children == None:
            self.front(node.right_children)
        # 返回最终结果
        return self.front_list
    # 中序(先左再中最后右)
    def middle(self,node=None):
        if node == None:
            node = self.root
        # 先判断左枝
        if not node.left_children == None:
            self.middle(node.left_children)
        # 输出当前节点
        self.middle_list.append(node.value)
        # 再判断右枝
        if not node.right_children == None:
            self.middle(node.right_children)
        return self.middle_list
    # 后序(先左再右最后中)
    def after(self,node=None):
        if node == None:
            node = self.root
        # 先判断左枝
        if not node.left_children == None:
            self.after(node.left_children)
        # 再判断右枝
        if not node.right_children == None:
            self.after(node.right_children)
        self.after_list.append(node.value)
        return self.after_list

    # 搜索
    def search(self,v,node=None):
        if node == None:
            node = self.root
        if node.value == v:
            return True
        if v > node.value:
            if not node.right_children == None:
                return self.search(v, node.right_children)
        else:
            if not node.left_children == None:
                return self.search(v, node.left_children)
        return False
if __name__ == __main__:
    # 需要建立二叉树的列表
    list = [4, 6, 3, 1, 7, 9, 8, 5, 2]
    t = tree()
    t.create_tree(0,list)
    res = t.front()
    print(前序,res)
    res = t.middle()
    print(中序,res)
    res = t.after()
    print(后序,res)
    i = t.search(8)
    print(8是否在二叉树中,i)

  中心就是递归判断传值,递归取值,很简单,下一步就是优化二叉树,二叉树的左旋,右旋,实现平衡二叉树。

      

以上是关于python 二叉树实现的主要内容,如果未能解决你的问题,请参考以下文章

Python数据结构系列☀️《树与二叉树-基础知识》——知识点讲解+代码实现☀️

python代码实现二叉树的镜像树

python 二叉树实现

python代码实现二叉树中最低的公共祖先

python实现二叉树的遍历以及基本操作

Python教程讲解二叉树的三种深度