在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()

Python把給定的列表轉化成二叉樹