二叉树的深度优先递归、非递归及广度优先遍历算法的python实现

一、二叉树遍历介绍

介绍

• 所谓遍历是指沿着某条搜索路线，依次对树中每个结点均做一次且仅做一次访问。访问结点所做的操作依赖于具体的应用问 题。 遍历是二叉树上最重要的运算之一，是二叉树上进行其它运算之基础。二叉树的遍历分为深度优先和广度优先两种，其中深度优先又包括前序遍历、中序遍历、后序遍历三种，所谓前、中、后是根据根节点与左右子树的遍历顺序决定的。前序遍历是先访问根节点再访问左子树最后访问右子树；中序遍历是先访问左子树再访问根节点最后访问右子树；后序遍历是先访问左子树再访问右子树最后访问根节点。广度优先遍历则是从根节点开始由上到下按层访问，每一层从左到右依次访问。下面分别介绍这几种遍历方式以及它们的代码实现。

二叉树结点定义

• python代码定义二叉树结点如下：

Class TreeNode(object):
   def  __init__(self,val):
   	self.lchild=None
   	self.rchild=None
   	self.val=val

前（先）序遍历

• 递归思想实现先序遍历较易理解，这里不做过多解释。
• python代码实现如下：

 def preOrder(self,root):
    if rooot == None:
        return
    print(root.val)
    self.preOrder(root.lchild)
    self.preOrder(root.rchild)

• 非递归遍历思路如下：
  1）定义一个栈
  2）将根节点压入栈中
  3）每次从栈中弹出结点记为cur并打印，如果右孩子不为空，将右孩子压入栈中，如果左孩子不为空，将左孩子压入栈中
  4）不断重复步骤3，直到栈空结束
• python代码实现如下：

def preOrder(self,root):
    if root==None:
        return
    stack=[]
    stack.append(root)
    while stack:
        cur = stack.pop()
        print(cur.val)
        if cur.rchild!=None:
            stack.append(stack.rchild)
        if cur.lchild!= None:
            stack.append(stack.lchild)

中序遍历

• 递归思想实现中序遍历较易理解，这里不做过多解释。
• python代码实现如下：

def inOrder(self,root):
    if root == None:
    return
    self.inOrder(root.lchild)
    print(root.val)
    self.inOrder(root.rchild)

• 非递归遍历思路如下：
  1）申请一个栈stack，申请一个变量cur指向头结点
  2）先把头结点压入栈中，对于以cur为头结点的整颗子树来说，依次把整颗树的左边界压入栈中，即不断令cur=cur.left
  3）不断重复步骤2，直到cur为空，此时从栈中弹出结点node并打印，然cur=node.right,然后重复步骤2
  4）当stack为空并且cur为空时，结束
• python代码实现如下：

def inOrder(self,root):
    if root == None:
        return
    stcak =[]
    cur = root
    stack.append(cur)
    while stcak:
        while cur.lchild:
             cur = cur.lchild
            stack.append(cur)
        node = stack.pop()
        print(node.val) 
        if node.rchild:     
            cur = node.rchild
            stack.append(cur)

后序遍历

• 递归思想实现先序遍历较易理解，这里不做过多解释。
• python代码实现如下：

def postOrder(self,root):
    if root == None:
        return
    self.postOrder(root.lchild)
    self.postOrder(root.rchild)
    print(root.val)

• 非递归遍历思路如下：
  1）申请两个栈s1，s2，然后将头结点压入s1
  2）从s1中弹出的结点记为cur,并加入到栈s2中，然后把cur的左孩子压入s1中，然后把cur的右孩子压入s1中
  3）不断重复2，直到s1为空
  4）从s2弹出结点并打印
• python代码实现如下：

def postOrder(self, root):
    if root == None:
        return
    stack1=[root.val]
    stack2=[]
    while stack1:
        cur = stack1.pop()
        stack2.append(cur)
        if cur.left:
            stack1.append(cur.left)
        if cur.right:
            stcak1.append(cur.right)
    while stack2:
        print(stack2.pop().val)

广度优先（层序）遍历

• 采用队列实现
• python代码实现如下：

def levelOrder(self,root):
    if root == None:
        return
     queue=[]
     queue.append(root)
     while queue；
         node = queue.pop(0)
         print(node.val)
         if node.lchild != None:
             queue.append(node.lchilid)
        if node.rchild != None:
             queue.append(node.rchild)