Python--实现二叉树的遍历操作

最新推荐文章于 2024-07-16 08:55:46 发布

Constantdropping

最新推荐文章于 2024-07-16 08:55:46 发布

阅读量259

点赞数

CC 4.0 BY-SA版权

分类专栏： python 数据结构文章标签：二叉树遍历

本文链接：https://blog.youkuaiyun.com/Constantdropping/article/details/98057125

python 同时被 2 个专栏收录

17 篇文章

订阅专栏

数据结构

7 篇文章

订阅专栏

本文详细介绍了二叉树的构建方法，并深入探讨了前序、中序、后序及层次遍历的递归与迭代实现，为读者提供了全面的二叉树操作指南。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、首先二叉树的定义：

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

构建一棵二叉树：

class Node(object):
    def __init__(self, val):
        self.val = val
        self.lchild = None
        self.rchild = None


class Tree(object):

    def __init__(self):
        self.root = None
        self.myQueue = []

    def add(self, val):
        node = Node(val)

        if self.root == None:
            self.root = node
            self.myQueue.append(self.root)
        else:
            while True:
                point = self.myQueue[0]

                if point.lchild == None:
                    point.lchild = node
                    self.myQueue.append(point.lchild)
                    return
                elif point.rchild == None:
                    point.rchild = node
                    self.myQueue.append(point.rchild)
                    self.myQueue.pop(0)
                    return

二、前序遍历

递归实现

def preorder(root,res=[]):
  if not root:
    return 
  res.append(root.val)
  preorder(root.lchild,res)
  preorder(root.rchild,res)
  return res

迭代实现

def preorder(root):
  res = []
  if not root: 
    return []
  stack = [root]
  while stack:
    node = stack.pop()
    res.append(node.val)
    if node.rchild:
      stack.append(node.rchild)
    if node.lchild:
      stack.append(node,lchild)
  return res

三、中序遍历

递归实现

def inorder(root,res=[]):
  if not root:
    return 
  inorder(root.lchild,res)
  res.append(root.val)
  inorder(root.rchild,res)
  return res


迭代实现

def inorder(root):
  stack = []
  node = root
  res = []
  while stack or node:
    while node:
      stack.append(node)
      node = node.lchild
    node = stack.pop()
    res.append(node.val)
    node = node.rchild
  return res

四、后序遍历

递归实现

def laorder(root,res=[]):
  if not root:
    return 
  laorder(root.lchild,res)
  laorder(root.rchild,res)
  res.append(root.val)
  return res



迭代实现

def laorder(root):
  stack = [root]
  res = []
  while stack:
    node = stack.pop()
    if node.lchild:
      stack.append(node.left)
    if node.rchild:
      stack.append(node.right)
    res.append(node.val)
  return res[::-1]

五、层次遍历

迭代

def levelorder(root):
  queue = [root]
  res = []
  while queue:
    node = queue.pop(0)
    if node.lchild: 
      queue.append(node.lchild)
    if node.rchild:
      queue.append(node.rchild)
    res.append(node.val)
  return res