Python 二叉数的实例化及遍历

原创

于 2024-06-02 11:38:27 发布 · 304 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #python

首先创建一个这样的二叉树，作为我们今天的实例。实例代码在下方。

#创建1个树类型
class TreeNode:
    def __init__(self,val,left=None,right=None):
        self.val=val
        self.left=left
        self.right=right
#实例化类
node1=TreeNode(5)
node2=TreeNode(6)
node3=TreeNode(7)
node4=TreeNode(1)
node5=TreeNode(2)
node1.left=node2
node1.right=node3
node2.left=node4
node2.right=node5

#指定根节点
root=node1

接下来我们便可以对上面创建好的树进行遍历了，下面是用递归的方法进行前序遍历和后序遍历。

class Solution:

    def preorder(self,root):
        self.result=[]
        self.dfs_pre(root)
        return self.result
    def postorder(self,root):
        self.result=[]
        self.dfs_post(root)
        return self.result

    def dfs_pre(self,root):
        if not root:
            return []
        self.result.append(ro