高频刷题-二叉树广度优先搜索(Breath First Search)专题

今天我们介绍二叉树的BFS和DFS。以前二叉树专题的时候已经使用BFS和DFS算法去解决过二叉树的相关问题。这次我们再总结一下BFS和DFS算法的特点和模板。

因为BFS算法的特点，在解决二叉树按层级遍历问题时非常有用。

我们将使用一个队列(Queue)来跟踪的所有节点级别之前我们跳到下一个级别。这也意味着,算法的空间复杂度是O (N),其中N是节点在每一层级的最大数量。

       下面我们直接看题目:

https://leetcode.com/problems/binary-tree-level-order-traversal/

BFS算法流程:

因为我们每次遍历时，都是访问该节点下一层级的所有子节点，所以我们需要一个队列Queue去保存每一层的所有节点。这样算法的步骤为：

1. 将根节点root放到queue中去；
2. 反复访问queue直到队列为空；
3. 每一次遍历时，先计算该层级的节点数量，使用循环遍历该层级的每一个节点；
4. 每一次层级遍历时，从队列中出列一个节点，将它的值保存到array中去(做相关的business log的操作)；
5. 访问步骤4中出队列的节点的左、右子树，分别将他们放在队列中；
6. 如果队列queue不为空，重复3开始下一层级的遍历。

对应代码如下：

public List<List<Integer>> levelOrder(TreeNode root) {
        
        // iteration 解决方法，采用队列每次存储每一层的节点，再依次遍历
        List<List<Integer>> res = new ArrayList<>();
       	
        if (root == null) return res; // 当输入为[]时候，返回不是null而是 []
        
        Queue<TreeNode> q = new LinkedList<>();
        
        q