二叉树的层序遍历

bfs

在while队列不为空的循环里，再加一层循环，遍历当前层的所有节点，即当前队列里的所有元素

class Solution(object):
    def levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:
            return []
        queue = [root]
        res = []
        while queue:
            nodenum = len(queue) #取出当前队列的长度，就是当前层的节点个数            
            curlevel = [] #用一个新的数组来存当前层的值，达到分层的效果
            for _ in range(nodenum):#一次循环中遍历当前层的所有节点
                node = queue.pop(0) #pop要放在循环里面做
                curlevel.append(node.val)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            res.append(curlevel) #一层遍历完，加入结果列表
        return res

加入层数信息的bfs

使用队列，队列中存储节点时，多用一个变量来存储节点所在的层数，也就是在队列中存n个包含两个元素的列表，[[node1,level1],[node2,level2],[node3,level2]]，当队列不为空时，先取出队列头，如果结果列表的长度等于当前节点对应的层数，那么就是该层的第一个节点，那么就加上一个空列表用于装新层的节点（例如level=0，len(res)=0;level=0; 就加上一个空列表，len(res)=1）

注意，判断头节点为空，直接返回。

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        res = []
        if root == None:
            return res
        queue = [[root,0]]
        while queue != []:
            h = queue.pop(0)
            f = h[0]
            level = h[1]
            if len(res) == level:
                res.append([])
            res[level].append(f.val)
            if f.left != None:
                queue.append([f.left,level+1])
            if f.right != None:
                queue.append([f.right,level+1])
        return res