python实现二叉树的遍历


Posted in Python onDecember 11, 2017

本文实例为大家分享了python实现二叉树的遍历具体代码,供大家参考,具体内容如下

python实现二叉树的遍历

代码:

# -*- coding: gb2312 -*- 
 
class Queue(object): 
 
  def __init__(self): 
    self.q = [] 
 
  def enqueue(self, item): 
    self.q.append(item) 
 
  def dequeue(self): 
    # if self.q != []: 
    if len(self.q)>0: 
       return self.q.pop(0)        
    else: 
      return None 
 
  def length(self): 
    return len(self.q) 
 
  def isempty(self): 
    return len(self.q)==0 
 
class Stack(object): 
  def __init__(self): 
    self.s = [] 
 
  def push(self, item): 
    self.s.append(item) 
 
  def pop(self): 
    if self.s !=[]: 
      item = self.s.pop(-1) 
    else: 
      item = None 
    return item 
 
  def length(self): 
    return len(self.s) 
 
  def isempty(self): 
    return self.s == [] 
 
  def top(self): 
    return self.s[-1] 
 
class TreeNode(object): 
 
  def __init__(self, data, left=None, right=None): 
    self.data = data 
    self.left = left 
    self.right = right 
    self.visited = False 
 
  def setData(self, data): 
    self.data = data 
 
  def setLeft(self, left): 
    self.left = left 
 
  def setRight(self, right): 
    self.right = right 
 
  def visit(self): 
    print self.data, 
    self.visited = True 
 
  def deVisit(self): 
    self.visited = False 
 
class BinaryTree(object): 
  def __init__(self, root): 
    self.root = root 
 
  # 前序遍历(递归) 
  def freshVisit(self, node): 
    if node is not None: 
      node.deVisit() 
    if node.left: 
      self.freshVisit(node.left) 
    if node.right: 
      self.freshVisit(node.right) 
 
  # 前序遍历(递归) 
  def preOrder(self, node): 
    if node is not None: 
      node.visit() 
    if node.left: 
      self.preOrder(node.left) 
    if node.right: 
      self.preOrder(node.right) 
 
  # 中序遍历(递归) 
  def inOrder(self, node): 
    if node.left: 
      self.inOrder(node.left)       
    if node is not None: 
      node.visit()     
    if node.right: 
      self.inOrder(node.right) 
 
  # 后序遍历(递归) 
  def postOrder(self, node): 
    if node.left: 
      self.postOrder(node.left)  
    if node.right: 
      self.postOrder(node.right) 
    if node is not None: 
      node.visit()   
 
  # 递归遍历 
  def orderTraveral(self, type): 
    if type == 0: 
      self.preOrder(self.root) 
    elif type == 1:  
      self.inOrder(self.root) 
    elif type == 2: 
      self.postOrder(self.root) 
 
  # 前序遍历(非递归) 
  # 用到一个栈和一个队列 
  # 首先是根节点入栈,再循环出栈 
  # 出栈元素不为空,则访问 
  # 出栈元素有左孩子节点则入栈,如果有右孩子节点则入队列 
  # 出栈元素为空,则访问队列 
  # 队列也为空则结束循环,否则队列元素出队 
  # 访问出队元素,出队元素有左孩子节点则入栈,出队元素有右孩子节点则入队列 
  # 循环直到最后退出 
  def preOrderByNotRecursion(self): 
    s = Stack() 
    q = Queue() 
    q.enqueue(self.root) 
 
    while not s.isempty() or not q.isempty(): 
 
      if not q.isempty(): 
        item = q.dequeue() 
        item.visit() 
        if item.left: 
          q.enqueue(item.left)           
        if item.right: 
          s.push(item.right) 
      elif not s.isempty(): 
        item = s.pop() 
        item.visit() 
        if item.left: 
          q.enqueue(item.left) 
        if item.right: 
          s.push(item.right) 
 
  # 前序遍历(非递归) 
  # 用到一个栈 
  # 首先是根节点入栈,再循环出栈 
  # 栈顶元素不为空,则访问, 并置已访问标志 
  # 如栈顶元素有左孩子节点则入栈 
  # 若栈顶元素已访问,则出栈 
  # 出栈元素若有右孩子节点则入栈 
  # 循环直到栈无元素退出         
  def preOrderByNotRecursion2(self): 
    s = Stack() 
    s.push(self.root) 
 
    while not s.isempty(): 
      item = s.top()       
      if item.visited: 
        s.pop() 
        if item.right: 
          s.push(item.right) 
      else: 
        item.visit() 
        if item.left: 
          s.push(item.left) 
 
 
  # 中序遍历(非递归) 
  # 用到一个栈 
  # 先将根节点入栈,循环出栈 
  # 如果出栈元素有左孩子节点并且左孩子节点没有访问过则入栈 
  # 反之,则出栈并且访问;如果出栈元素有右孩子节点则入栈 
  # 重复以上循环直到栈为空 
  def inOrderByNotRecursion(self): 
    s = Stack() 
    s.push(self.root) 
 
    while not s.isempty(): 
      item = s.top() 
      while(item.left and not item.left.visited): 
        s.push(item.left) 
        item = item.left 
      else: 
        item = s.pop() 
        item.visit() 
        if item.right: 
          s.push(item.right) 
 
 
  # 后序遍历(非递归) 
  # 用到一个栈 
  # 先将根节点入栈,循环出栈 
  # 如果出栈元素有左孩子节点并且左孩子节点没有访问过则入栈 
  # 反之,如果栈顶元素如果有右孩子节点并且右孩子节点没有访问过,则入栈 
  # 否则,出栈并访问 
  # 重复以上循环直到栈为空 
  def postOrderByNotRecursion(self): 
    s = Stack() 
    s.push(self.root) 
 
    while not s.isempty(): 
      item = s.top() 
      while(item.left and not item.left.visited): 
        s.push(item.left) 
        item = item.left 
      else: 
        if item.right and not item.right.visited: 
          s.push(item.right) 
        else: 
          item = s.pop() 
          item.visit() 
 
 
  # 层次遍历(非递归) 
  # 用到一个队列 
  # 先将根节点入队列 
  # 从队列取出一个元素,访问 
  # 如有左孩子节点则入队,如有右孩子节点则入队 
  # 重复以上操作直到队列入空 
  def layerOrder(self): 
    q = Queue() 
    q.enqueue(self.root) 
 
    while not q.isempty(): 
      item = q.dequeue() 
      item.visit() 
      if item.left: 
        q.enqueue(item.left) 
      if item.right: 
        q.enqueue(item.right) 
 
#      A 
#    B      C 
#  D    E  F    G 
#H 
if __name__ == '__main__': 
 
  nE = TreeNode('E'); 
  nF = TreeNode('F'); 
  nG = TreeNode('G'); 
  nH = TreeNode('H'); 
  nD = TreeNode('D', nH); 
  nB = TreeNode('B', nD, nE); 
  nC = TreeNode('C', nF, nG); 
  nA = TreeNode('A', nB, nC); 
  bTree = BinaryTree(nA); 
 
  # 前序递归遍历 
  print '----------前序遍历(递归)-----------' 
  bTree.orderTraveral(0) 
  print '\n----------中序遍历(递归)-----------' 
  bTree.orderTraveral(1) 
  print '\n----------后序遍历(递归)-----------' 
  bTree.orderTraveral(2) 
 
  print '\n\n----------前序遍历(非递归)-----------' 
  print '----------方法一-----------' 
  bTree.freshVisit(bTree.root) 
  bTree.preOrderByNotRecursion() 
  print '\n----------方法二-----------' 
  bTree.freshVisit(bTree.root)   
  bTree.preOrderByNotRecursion2() 
  print '\n\n----------中序遍历(非递归)-----------' 
  bTree.freshVisit(bTree.root) 
  bTree.inOrderByNotRecursion() 
  print '\n\n----------后序遍历(非递归)-----------' 
  bTree.freshVisit(bTree.root) 
  bTree.postOrderByNotRecursion() 
 
  print '\n\n----------层次遍历(非递归)-----------' 
  bTree.freshVisit(bTree.root) 
  bTree.layerOrder()

结果:

python实现二叉树的遍历

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python中的列表推导浅析
Apr 26 Python
详解 Python中LEGB和闭包及装饰器
Aug 03 Python
python+mongodb数据抓取详细介绍
Oct 25 Python
Python中实例化class的执行顺序示例详解
Oct 14 Python
Python生成一个迭代器的实操方法
Jun 18 Python
python的sorted用法详解
Jun 25 Python
python 字典 setdefault()和get()方法比较详解
Aug 07 Python
python基于json文件实现的gearman任务自动重启代码实例
Aug 13 Python
Python3 中作为一等对象的函数解析
Dec 11 Python
Python基于unittest实现测试用例执行
Nov 25 Python
Python日志打印里logging.getLogger源码分析详解
Jan 17 Python
Python实现文本文件拆分写入到多个文本文件的方法
Apr 18 Python
django上传图片并生成缩略图方法示例
Dec 11 #Python
使用Python的package机制如何简化utils包设计详解
Dec 11 #Python
python timestamp和datetime之间转换详解
Dec 11 #Python
Python时间戳使用和相互转换详解
Dec 11 #Python
python的exec、eval使用分析
Dec 11 #Python
Python中eval带来的潜在风险代码分析
Dec 11 #Python
Python验证文件是否可读写代码分享
Dec 11 #Python
You might like
用PHP实现读取和编写XML DOM代码
2010/04/07 PHP
linux下实现定时执行php脚本
2015/02/13 PHP
PHP常用的三种设计模式
2017/02/17 PHP
PHP实现数组转JSon和JSon转数组的方法示例
2018/06/14 PHP
Javascript & DHTML 实例编程(教程)DOM基础和基本API
2007/06/02 Javascript
javascript 一段左右两边随屏滚动的代码
2009/06/18 Javascript
使用Grunt.js管理你项目的应用说明
2013/04/24 Javascript
判断javascript的数据类型(示例代码)
2013/12/11 Javascript
使用jquery.upload.js实现异步上传示例代码
2014/07/29 Javascript
基于jquery实现智能表单验证操作
2016/05/09 Javascript
js中获取 table节点各tr及td的内容简单实例
2016/10/14 Javascript
ES6使用let命令更简单的实现块级作用域实例分析
2017/03/31 Javascript
jQuery EasyUI结合zTree树形结构制作web页面
2017/09/01 jQuery
详解小程序原生使用ES7 async/await语法
2018/08/06 Javascript
实例详解vue中的$root和$parent
2019/04/29 Javascript
微信小程序云开发如何使用云函数生成二维码
2019/05/18 Javascript
小程序如何构建骨架屏
2019/05/29 Javascript
VUE写一个简单的表格实例
2019/08/06 Javascript
仅用50行Python代码实现一个简单的代理服务器
2015/04/08 Python
Python 26进制计算实现方法
2015/05/28 Python
Pythont特殊语法filter,map,reduce,apply使用方法
2016/02/27 Python
Python单体模式的几种常见实现方法详解
2017/07/28 Python
使用matplotlib画散点图的方法
2018/05/25 Python
python中时间模块的基本使用教程
2019/05/14 Python
python tkinter实现界面切换的示例代码
2019/06/14 Python
python tkinter图形界面代码统计工具(更新)
2019/09/18 Python
Python中关于浮点数的冷知识
2019/09/22 Python
使用Python下载抖音各大V视频的思路详解
2021/02/06 Python
CSS3 clip-path 用法介绍详解
2018/03/01 HTML / CSS
澳大利亚潮流尖端的快时尚品牌:Cotton On
2016/09/26 全球购物
贝尔帐篷精品店:Bell Tent Boutique
2019/06/12 全球购物
Python面试题集
2012/03/08 面试题
大课间活动实施方案
2014/03/06 职场文书
学校爱心捐款倡议书
2014/05/13 职场文书
电子专业求职信
2014/06/19 职场文书
2014年后备干部工作总结
2014/12/08 职场文书