Python实现二叉树的遍历

二叉树是有限个元素的集合，该集合或者为空、或者有一个称为根节点（root）的元素及两个互不相交的、分别被称为左子树和右子树的二叉树组成。

• 二叉树的每个结点至多只有二棵子树(不存在度大于2的结点)，二叉树的子树有左右之分，次序不能颠倒。
• 二叉树的第i层至多有2^{i-1}个结点
• 深度为k的二叉树至多有2^k-1个结点；
• 对任何一棵二叉树T，如果其终端结点数为N0，度为2的结点数为N2，则N0=N2+1
• 前序遍历（先序遍历）：根节点、左子树、右子树；
• 中序遍历：左子树、根节点、右子树；
• 后序遍历：左子树、右子树、根节点

首先构建二叉树：

class Node:
    def __init__(self, value = None, left = None, right = None):
        self.value = value
        self.left = left #左子树
        self.right = right #右子树

下面是3种遍历方式及测试代码：

def preTraverse(root):
    '''
    前序遍历
    '''
    if root == None:
        return
    print(root.value)
    preTraverse(root.left)
    preTraverse(root.right)


def midTraverse(root):
    '''
    中序遍历
    '''
    if root == None:
        return
    midTraverse(root.left)
    print(root.value)
    midTraverse(root.right)


def afterTraverse(root):
    '''
    后序遍历
    '''
    if root == None:
        return
    afterTraverse(root.left)
    afterTraverse(root.right)
    print(root.value)

if __name__=='__main__':
    root=Node('D',Node('B',Node('A'),Node('C')),Node('E',right=Node('G',Node('F'))))
    print('前序遍历：')
    preTraverse(root)
    print('\n')
    print('中序遍历：')
    midTraverse(root)
    print('\n')
    print('后序遍历：')
    afterTraverse(root)
    print('\n')

输出结果为：

前序遍历：
D
B
A
C
E
G
F

中序遍历：
A
B
C
D
E
F
G

后序遍历：
A
C
B
F
G
E
D

 

 

那么，如果我们已知二叉树的前序遍历和中序遍历，求这棵二叉树的后序遍历：

主要思路：

1、先序遍历第一个节点代表根节点root

2、根据root在中序遍历中的位置，把二叉树分成两部分，root左侧为左子树、root右侧为右子树。

3、找出中序遍历中左子树对应的先序遍历数字，第一个节点代表左子树的根节点。然后按此规律递归即可。

以下是转账文章中的代码，可能有错，后面还有一道根据后序和中序就先序的是牛客上的题目。

preList = list('12473568')
midList = list('47215386')
afterList = []

def findTree(preList, midList, afterList):
    if len(preList) == 0:
        return
    if len(preList) == 1:
        afterList.append(preList[0])
        return
    root = preList[0]
    n = midList.index(root)
    findTree(preList[1:n + 1], midList[:n], afterList)
    findTree(preList[n + 1:], midList[n + 1:], afterList)
    afterList.append(root)

结果为：

['7', '4', '2', '5', '8', '6', '3', '1']

1 如果以上面的前序：DBACEGF和中序：ABCDEFG，得到的结果为：
2 ['A', 'C', 'B', 'F', 'G', 'E', 'D']

 

 牛客网题目：https://ac.nowcoder.com/acm/problem/16692

给出一棵二叉树的中序与后序排列。求出它的先序排列。（约定树结点用不同的大写字母表示，长度 ≤ 8）

midList = list(input())
afterList = list(input())
preList = []

def findTree(afterList, midList, preList):
    if len(afterList) == 0:
        return

    if len(afterList) == 1:  # 如果后续遍历长度为1，则先序遍历就是后续遍历
        preList.append(afterList[0])
        return
    root = afterList[-1]  # 根节点始终是后续遍历最后一个
    n = midList.index(root)
    preList.append(root) #找到root节点在中序遍历中的位置，根据中序遍历将树分成左子树和右子树
    findTree(afterList[0:n], midList[:n], preList) # 处理左子树
    findTree(afterList[n:-1], midList[n + 1:], preList) # 处理右子树

findTree(afterList, midList, preList)
for p in preList:
    print(p, end='')