首页 > 编程 > Python > 正文

Python利用前序和中序遍历结果重建二叉树的方法

2020-01-04 17:30:41
字体:
来源:转载
供稿:网友
这篇文章主要介绍了Python利用前序和中序遍历结果重建二叉树的方法,实例分析了Python二叉树的定义与遍历操作技巧,需要的朋友可以参考下
 

本文实例讲述了Python利用前序和中序遍历结果重建二叉树的方法。分享给大家供大家参考,具体如下:

题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

这道题比较容易,前序遍历的结果中,第一个结点一定是根结点,然后在中序遍历的结果中查找这个根结点,根结点左边的就是左子树,根结点右边的就是右子树,递归构造出左、右子树即可。示意图如图所示:

Python,二叉树

利用前序和中序遍历的结果重建二叉树

Python代码:

# coding: utf-8'''题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。'''class Node:  def __init__(self, data, left, right):    self.data = data    self.left = left    self.right = rightdef construct_tree(pre_order, mid_order):  # 忽略参数合法性判断  if len(pre_order) == 0 :    return None  # 前序遍历的第一个结点一定是根结点  root_data = pre_order[0]  for i in range(0, len(mid_order)):    if mid_order[i] == root_data:      break  # 递归构造左子树和右子树  left = construct_tree(pre_order[1 : 1 + i], mid_order[:i])  right = construct_tree(pre_order[1 + i:], mid_order[i+1:])  return Node(root_data, left, right)if __name__ == '__main__':  pre_order = [1, 2, 4, 7, 3, 5, 6, 8]  mid_order = [4, 7, 2, 1, 5, 3, 8, 6]  root = construct_tree(pre_order, mid_order)  print root.data  print root.left.data  print root.right.data  print root.left.left.data  print root.left.left.right.data  print root.right.right.left  print root.right.left.data

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表