首页 > 编程 > Python > 正文

Python实现二叉树的常见遍历操作总结【7种方法】

2020-01-04 13:35:24
字体:
来源:转载
供稿:网友

本文实例讲述了Python实现二叉树的常见遍历操作。分享给大家供大家参考,具体如下:

二叉树的定义:

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

二叉树的前序遍历

递归

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

迭代

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

二叉树的中序遍历

递归

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

迭代

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

二叉树的后序遍历

递归

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

迭代

def laorder(root):  stack=[root]  res=[]  while stack:    node=stack.pop()    if node.left:      stack.append(node.left)    if node.right:      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.left:       queue.append(node.left)    if node.right:      queue.append(node.right)    res.append(node.val)  return res

 

希望本文所述对大家Python程序设计有所帮助。


注:相关教程知识阅读请移步到python教程频道。
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表