首先进行二叉树的结点设计:
class TreeNode(object):
# 先写value,有的不一定有左右子树
def __init__(self, value, left_child=None, right_child=None):
self.left_child = left_child
self.value = value
self.right_child = right_child
设计一个二叉树,结构如图:
代码如下:
node_q = TreeNode('Q')
node_k = TreeNode('K', node_q)
node_g = TreeNode('G')
node_b = TreeNode('B', node_g, node_k)
node_f = TreeNode('F')
node_a = TreeNode('A', node_b, node_f)
注意:从下往上设计关系
深度遍历:(其核心是 递归)
def deep_search(root):
print(root.value)
if root.left_child:
deep_search(root.left_child)
if root.right_child:
deep_search(root.right_child)
deep_search(node_a)
广度遍历:
def wide_search(root):
queue = [root]
while queue:
node = queue.pop(0)
print(node.value)
if node.left_child:
queue.append(node.left_child)
if node.right_child:
queue.append(node.right_child)
wide_search(node_a)