Python二叉树的定义及常用遍历算法分析


Posted in Python onNovember 24, 2017

本文实例讲述了Python二叉树的定义及常用遍历算法。分享给大家供大家参考,具体如下:

说起二叉树的遍历,大学里讲的是递归算法,大多数人首先想到也是递归算法。但作为一个有理想有追求的程序员。也应该学学非递归算法实现二叉树遍历。二叉树的非递归算法需要用到辅助栈,算法着实巧妙,令人脑洞大开。

以下直入主题:

定义一颗二叉树,请看官自行想象其形状,

class BinNode( ):
  def __init__( self, val ):
    self.lchild = None
    self.rchild = None
    self.value = val
binNode1 = BinNode( 1 )
binNode2 = BinNode( 2 )
binNode3 = BinNode( 3 )
binNode4 = BinNode( 4 )
binNode5 = BinNode( 5 )
binNode6 = BinNode( 6 )
binNode1.lchild = binNode2
binNode1.rchild = binNode3
binNode2.lchild = binNode4
binNode2.rchild = binNode5
binNode3.lchild = binNode6

先序遍历:

'''
先序遍历二叉树
'''
def bin_tree_pre_order_traverse( root, visit_func ):
  s = Stack()
  s.push( root )
  while not s.is_empty():
    node = s.pop()
    visit_func( node )
    if node.rchild:
      s.push( node.rchild )
    if node.lchild:
      s.push( node.lchild )

中序遍历:

'''
中序遍历二叉树
'''
def bin_tree_in_order_traverse( root, visit_func ):
  s = Stack()
  node = root
  while node or not s.is_empty():
    if node:
      s.push( node )
      node = node.lchild
    else:
      node = s.pop()
      visit_func( node )
      node = node.rchild

后序遍历:

后序遍历中,要保证左孩子和右孩子都已被访问才能访问根结点,并且左孩子需在右孩子前访问,这就为流程的控制带来了难题。下面介绍两种思路。

思路一,双栈法,这种方式比较容易理解,缺点是需要两个栈。

'''
后序遍历二叉树
'''
def bin_tree_post_order_traverse( root, visit_func ):
  s1 = Stack()
  s2 = Stack()
  s1.push( root )
  while not s1.is_empty():
    node = s1.pop()
    s2.push( node )
    if node.lchild:
      s1.push( node.lchild )
    if node.rchild:
      s1.push( node.rchild )
  while not s2.is_empty():
    visit_func( s2.pop() )

思路二,要保证根结点在左孩子和右孩子访问之后才能访问,因此对于任一结点P,先将其入栈。如果P不存在左孩子和右孩子,则可以直接访问它;或者P存在左孩子或者右孩子,但是其左孩子和右孩子都已被访问过了,则同样可以直接访问该结点。若非上述两种情况,则将P的右孩子和左孩子依次入栈,这样就保证了每次取栈顶元素的时候,左孩子在右孩子前面被访问,左孩子和右孩子都在根结点前面被访问。

def bin_tree_post_order_traverse2( root, visit_func ):
  curr = root
  prev = None
  s = Stack()
  s.push( curr )
  while not s.is_empty():
    curr = s.peek()
    if ( not curr.lchild and not curr.rchild ) or ( prev and ( prev == curr.lchild or prev == curr.rchild ) ):
      visit_func( curr )
      s.pop()


 prev = curr
    else:
      if curr.rchild:
        s.push( curr.rchild )
      if curr.lchild:
        s.push( curr.lchild )

层序遍历:

def bin_tree_level_traverse( root, visit_func ):
  queue = Queue()
  queue.enqueue( root )
  while not queue.is_empty():
    node = queue.dequeue().value
    visit_func( node )
    if node.lchild:
      queue.enqueue( node.lchild )
    if node.rchild:
      queue.enqueue( node.rchild )

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

Python 相关文章推荐
python笔记(1) 关于我们应不应该继续学习python
Oct 24 Python
Python中的Matplotlib模块入门教程
Apr 15 Python
python在linux系统下获取系统内存使用情况的方法
May 11 Python
快速了解Python中的装饰器
Jan 11 Python
python爬虫获取百度首页内容教学
Dec 23 Python
python f-string式格式化听语音流程讲解
Jun 18 Python
python使用sessions模拟登录淘宝的方式
Aug 16 Python
python利用openpyxl拆分多个工作表的工作簿的方法
Sep 27 Python
Pytorch 实现sobel算子的卷积操作详解
Jan 10 Python
pyinstaller 3.6版本通过pip安装失败的解决办法(推荐)
Jan 18 Python
VS2019+python3.7+opencv4.1+tensorflow1.13配置详解
Apr 16 Python
python中如何设置代码自动提示
Jul 15 Python
详解python上传文件和字符到PHP服务器
Nov 24 #Python
Python实现矩阵转置的方法分析
Nov 24 #Python
利用Django内置的认证视图实现用户密码重置功能详解
Nov 24 #Python
利用python3随机生成中文字符的实现方法
Nov 24 #Python
Python内置函数 next的具体使用方法
Nov 24 #Python
python中WSGI是什么,Python应用WSGI详解
Nov 24 #Python
python中print()函数的“,”与java中System.out.print()函数中的“+”功能详解
Nov 24 #Python
You might like
表单复选框向PHP传输数据的代码
2007/11/13 PHP
解析如何屏蔽php中的phpinfo()函数
2013/06/06 PHP
ThinkPHP设置禁止百度等搜索引擎转码(简单实用)
2016/02/15 PHP
PHP实现通过URL提取根域名
2016/03/31 PHP
PHP自定义函数格式化json数据示例
2016/09/14 PHP
微信小程序发送订阅消息的方法(php 为例)
2019/10/30 PHP
整理8个很棒的 jQuery 倒计时插件和教程
2011/12/12 Javascript
location.href用法总结(最主要的)
2013/12/27 Javascript
js 采用delete实现继承示例代码
2014/05/20 Javascript
javascript实现瀑布流加载图片原理
2016/02/02 Javascript
jquery获取input type=text中的值的各种方式(总结)
2016/12/02 Javascript
原生js和css实现图片轮播效果
2017/02/07 Javascript
详解node.js搭建代理服务器请求数据
2017/04/08 Javascript
页面缩放兼容性处理方法(zoom,Firefox火狐浏览器)
2017/08/29 Javascript
微信小程序实现分享朋友圈的图片功能示例
2019/01/18 Javascript
js中对象与对象创建方法的各种方法
2019/02/27 Javascript
javascript设计模式 ? 简单工厂模式原理与应用实例分析
2020/04/09 Javascript
js实现随机圆与矩形功能
2020/10/29 Javascript
[05:00]TI9战队采访 - Royal Never Give Up
2019/08/20 DOTA
python方向键控制上下左右代码
2018/01/20 Python
Python实现k-means算法
2018/02/23 Python
Python FTP两个文件夹间的同步实例代码
2018/05/25 Python
python 弹窗提示警告框MessageBox的实例
2019/06/18 Python
python下载库的步骤方法
2019/10/12 Python
解决python 上传图片限制格式问题
2019/10/30 Python
python pprint模块中print()和pprint()两者的区别
2020/02/10 Python
Python 列表中的修改、添加和删除元素的实现
2020/06/11 Python
浅谈pytorch中的BN层的注意事项
2020/06/23 Python
pandas参数设置的实用小技巧
2020/08/23 Python
scrapy处理python爬虫调度详解
2020/11/23 Python
英国第一蛋白粉品牌:Myprotein
2016/09/14 全球购物
成教毕业生自我鉴定
2013/10/23 职场文书
大三在校生电子商务求职信
2013/10/29 职场文书
绍兴鲁迅故居导游词
2015/02/09 职场文书
Python实现打乒乓小游戏
2021/09/25 Python
抖音动画片,皮皮虾,《治愈系》动画在用这首REMIX作为背景音乐,Anak ,The last world with you完整版
2022/03/16 杂记