面试题32 - I. 从上到下打印二叉树

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回：

[3,9,20,15,7]

思路：该题用到BFS思想，其实跟BFS还是有些差别；因为BFS需要考虑邻接节点有没有被遍历。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def levelOrder(self, root: TreeNode) -> List[int]:
        if not root: return []
        res = []
        q = [root]
        while q:
            node = q.pop(0)
            res.append(node.val)
            if node.left: q.append(node.left)
            if node.right: q.append(node.right)
        return res

下面是BFS的代码：

def BFS(graph, s):#graph图，s为初始节点
    queue = []#队列，先进先出。先放进去再弹出来
    queue.append(s)
    visited = set()#空集合，利用集合的唯一性质，遍历过的节点都存储其中
    visited.add(s)
    while(len(queue) > 0):
        vertex = queue.pop(0)
        nodes = graph[vertex]#bfs取出节点对应的邻接节点
        for w in nodes:
            if w not in visited:
                queue.append(w)
                visited.add(w)
        print(vertex)