Python二叉树遍历算法

# 定义一棵树结构
class Tree:
    def __init__(self, value):
        self.value = value
        self.left = None
        self.right = None


def zhong(tree, array=[]):
    """
    中序遍历
    :param tree: 
    :param array: 
    :return: 
    """
    if tree is None:
        return array
    zhong(tree.left, array)
    array.append(tree.value)
    zhong(tree.right, array)
    return array


def pre(tree, array=[]):
    """
    前序遍历
    :param tree: 
    :param array: 
    :return: 
    """
    if tree is None:
        return array
    array.append(tree.value)
    pre(tree.left, array)
    pre(tree.right, array)
    return array


def behind(tree, array=[]):
    """
    后序遍历
    :param tree: 
    :param array: 
    :return: 
    """
    if tree is None:
        return array
    behind(tree.left, array)
    behind(tree.right, array)
    array.append(tree.value)
    return array


# 按照给出的树,构建树形结构
tree = Tree(10)
tree.left = Tree(5)
tree.right = Tree(15)
tree.left.left = Tree(2)
tree.left.right = Tree(5)
tree.left.left.left = Tree(1)
tree.right.right = Tree(22)

print(zhong(tree, []))
print(behind(tree, []))
print(pre(tree, []))

参考视频:https://www.bilibili.com/video/BV1Ev411i7Rk/?spm_id_from=333.337.search-card.all.click&vd_source=30aff718c4a81a6d17f211397fc39bb7

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
二叉树遍历算法是指按照某种顺序访问二叉树中的所有节点。常见的遍历方式有三种:前序遍历、中序遍历和后序遍历。其中,前序遍历是指先访问根节点,然后按照左子树、右子树的顺序递归遍历子树;中序遍历是指先递归遍历左子树,然后访问根节点,最后递归遍历右子树;后序遍历是指先递归遍历左子树和右子树,最后访问根节点。 二叉树遍历算法的设计思路可以归纳为以下几个步骤: 1. 如果当前节点为空,则返回。 2. 访问当前节点。 3. 递归遍历左子树。 4. 递归遍历右子树。 具体实现时,可以采用递归或者非递归的方式。递归方式比较简单,但是可能会导致栈溢出。非递归方式可以使用栈来模拟递归过程,避免栈溢出的问题。 以下是三种遍历方式的Python代码实现: 1.前序遍历 ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right class Solution: def preorderTraversal(self, root: TreeNode) -> List[int]: if not root: return [] res = [] res.append(root.val) res += self.preorderTraversal(root.left) res += self.preorderTraversal(root.right) return res ``` 2.中序遍历 ```python class Solution: def inorderTraversal(self, root: TreeNode) -> List[int]: if not root: return [] res = [] res += self.inorderTraversal(root.left) res.append(root.val) res += self.inorderTraversal(root.right) return res ``` 3.后序遍历 ```python class Solution: def postorderTraversal(self, root: TreeNode) -> List[int]: if not root: return [] res = [] res += self.postorderTraversal(root.left) res += self.postorderTraversal(root.right) res.append(root.val) return res ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值