zoukankan      html  css  js  c++  java
  • Python把给定的列表转化成二叉树

            在LeetCode上做题时,有很多二叉树相关题目的测试数据是用列表给出的,提交的时候有时会出现一些数据通不过,这就需要在本地调试,因此需要使用列表来构建二叉树,方便自己调试。LeetCode上二叉树结点的定义如下:

    1 class TreeNode(object):
    2     def __init__(self, x):
    3         self.val = x
    4         self.left = None
    5         self.right = None

           使用列表构建二叉树,以及二叉树的层次遍历,先序遍历,中序遍历,后序遍历的代码如下所示:

     1 from collections import deque
     2 
     3 
     4 class Tree(object):
     5     def __init__(self):
     6         self.root = None
     7     
     8     def construct_tree(self, values=None):
     9         if not values:
    10             return None
    11         self.root = TreeNode(values[0])
    12         queue = deque([self.root])
    13         leng = len(values)
    14         nums = 1
    15         while nums < leng:
    16             node = queue.popleft()
    17             if node:
    18                 node.left = TreeNode(values[nums]) if values[nums] else None
    19                 queue.append(node.left)
    20                 if nums + 1 < leng:
    21                     node.right = TreeNode(values[nums+1]) if values[nums+1] else None
    22                     queue.append(node.right)
    23                     nums += 1
    24                 nums += 1
    25     
    26     def bfs(self):
    27         ret = []
    28         queue = deque([self.root])
    29         while queue:
    30             node = queue.popleft()
    31             if node:
    32                 ret.append(node.val)
    33                 queue.append(node.left)
    34                 queue.append(node.right)
    35         return ret
    36 
    37     def pre_traversal(self):
    38         ret = []
    39 
    40         def traversal(head):
    41             if not head:
    42                 return
    43             ret.append(head.val)
    44             traversal(head.left)
    45             traversal(head.right)
    46         traversal(self.root)
    47         return ret
    48 
    49     def in_traversal(self):
    50         ret = []
    51 
    52         def traversal(head):
    53             if not head:
    54                 return
    55             traversal(head.left)
    56             ret.append(head.val)
    57             traversal(head.right)
    58 
    59         traversal(self.root)
    60         return ret
    61 
    62     def post_traversal(self):
    63         ret = []
    64 
    65         def traversal(head):
    66             if not head:
    67                 return
    68             traversal(head.left)
    69             traversal(head.right)
    70             ret.append(head.val)
    71 
    72         traversal(self.root)
    73         return ret

              测试以及使用:

    1 t = Tree()
    2 t.construct_tree([1, 2, None, 4, 3, None, 5])
    3 print t.bfs()
    4 print t.pre_traversal()
    5 print t.in_traversal()
    6 print t.post_traversal()
  • 相关阅读:
    回顾[2007-09-03 12:58:03]
    关于知音[2007-08-17 20:56:06]
    今天晚上吃散伙饭[2007-06-18 00:24:36]
    上次所料不错[2007-06-13 15:44:47]
    今天真没劲[2007-06-10 17:50:25]
    关于昨晚的梦[2007-05-07 12:12:06]
    iOS 自定义键盘
    iOSQuart2D绘图之UIImage简单使用
    iOS 两种不同的图片无限轮播
    iOS 简单引导界面
  • 原文地址:https://www.cnblogs.com/qiaojushuang/p/7724936.html
Copyright © 2011-2022 走看看