【剑指 Offer 32 - II. 从上到下打印二叉树 II】

题目描述：

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shun-shi-zhen-da-yin-ju-zhen-lcof/

• 从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
• 即：给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

示例：

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

				    3
				   / \
				  9  20
				    /  \
				   15   7

• 返回其层次遍历结果：

				[
				  [3],
				  [9,20],
				  [15,7]
				]

• 示例2：

输入：root = [1]
输出：[[1]]

• 示例3：

输入：root = []
输出：[]

• 提示：

	树中节点数目在范围 [0, 2000] 内
	-1000 <= Node.val <= 1000

解析思路：

• 参考k神解析.

• 题目要求将二叉树从上到下，按照层打印，又称为二叉树的 广度优先搜索（BFS）；BFS 通常借助 队列的先入先出特点实现。

• 逻辑方法：循环将二叉树每一层节点打印到一行，并且在利用取到的当前层节点，将下一层全部节点加入 队列，本层的节点数暂存在 temp=[ ] 中 ，以此循环，直到最后一层，队列中无元素跳出循环；每次循环结束，将temp=[ ]中本层暂存的节点 添加到最终的结果 res = [ ] 中。

• 最后：返回值 res = [ ] ；

代码（python3）

# 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[List[int]]:
        if not root : return []
        res = []
        queue = collections.deque() # 使用Python的双端队列
        queue.append(root)     # 添加首个节点
        while queue: # 当队列 queue 为空时跳出
            temp = []
            for _ in range(len(queue)): 
                node = queue.popleft()  # 从双端队列的左边取出
                temp.append(node.val)   # 将该层节点加入独立的列表中
                if node.left : queue.append(node.left)  # 如果该节点有下一层子节点，将下层子节点添加如队列
                if node.right:queue.append(node.right)
            res.append(temp) # 每层取出的单独列表放到总列表里打印
        return res
        

• 补充：

• collections是python自带的库，collections.deque()是双端队列，可以实现左右两端添加或取出元素的功能，时间复杂度均为O(1)，同时具有栈和队列的性质。

• 使用示例： append与extend方法的区别:

q = collections.deque()
q.append('a') # 在队列右边添加元素‘a’，deque(['a'])
q.appendleft('b') # 在队列左边添加元素‘b’，deque(['b', 'a'])
q.extend('c') # 在队列右边添加序列，deque(['b', 'a', 'c])
q.extendleft('d') # 在队列左边添加序列，deque(['d', 'b', 'a', 'c'])
# 关于append和extend的区别
# append添加一整个对象，不管该对象是什么类型，都作为一个整体全部添加进来，如整型数据或者一整个列表
# extend添加一个序列，只能是相同类型的元素，如该题中的字符/字符串或者是由字符和字符串组成的列表['st','e']，其他类型的数据如整数就会报错

 # 实例：

q.append(['e','f'] # deque(['d', 'b', 'a', 'c', ['e', 'f']])
q.append(4) # deque(['d', 'b', 'a', 'c', ['e', 'f'], 4])
q.extend([g', 'h'] # deque(['d', 'b', 'a', 'c', ['e', 'f'], 4, 'g', 'h'])
q.append(5) # 报错 TypeError: 'int' object is not iterable
# 取出元素
q.pop() # 右端取出
q.popleft() # 左端取出

• 详细解析参考： k神- jyd 解析.

代码（cpp）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> q;   // 声明一个队列变量
        vector<vector<int>> res;   // 声明一个vector 变量
        if (root == NULL) return res;
        q.push(root);      // 元素压入队列  push  
        while(!q.empty()){
            vector<int> temp;
            for(int i=q.size();i>0;i--){  // 队列，元素先进先出规则 （倒序）
                TreeNode* node = q.front();    // 取出栈中的节点 
                q.pop();      
                temp.push_back(node->val);   // 元素压入 vector  push_back 
                if (node->left!=NULL) q.push(node->left); 
                if (node->right != NULL) q.push(node->right);
            }
            res.push_back(temp);
        }
        return res;
    }
};

复杂度分析：

• 时间复杂度 O(N) ： N 为二叉树的节点数量，即 BFS 需循环 N 次。
• 空间复杂度 O(N) ： 最差情况下，即当树为平衡二叉树时，最多有 N/2 个树节点同时在 queue 中，使用 O(N) 大小的额外空间。
• 平衡二叉树图：也称为 AVL 树
• 可以是空树。
• 假如不是空树，任何一个结点的左子树与右子树都是平衡二叉树，并且高度之差的绝对值不超过 1。