python 深度、广度遍历二叉树

首先进行二叉树的结点设计:

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)

猜你喜欢

转载自blog.csdn.net/weixin_42336579/article/details/81334212