二叉树的层序遍历实现

---

题目

给定一个二叉树，返回该二叉树层序遍历的结果，（从左到右，一层一层地遍历）
例如：
给定的二叉树是 {3,9,20,#,#,15,7}；

该二叉树层序遍历的结果是：
[
[3],
[9,20],
[15,7]
]

---

思路

层序遍历可以考虑通过队列或者栈来实现。

若是队列实现，则每次需要加入当前层的所有结点到队头，之后收集完当前层的所有结点数值后，在加入下一层的所有结点，直至叶子结点。

队列实现流程

• 维护一个队列Queue，队列负责保存每一层的所有结点。
• List集合负责收集当前层的所有node结点的数值value。
• 遍历所有层，从queue队列中取出当前层的所有结点，并收集结果res到集合中。
• 左右结点按照顺序添加到队尾。

---

代码实现

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     *
     * @param root TreeNode类
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
        // write code here
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        //利用队列保存每一层的结点
        Queue<TreeNode> queue = new LinkedList<>();
        //首先放入根节点
        queue.offer(root);
        while (!queue.isEmpty()) {
            //收集当前层的所有结点数值
            ArrayList<Integer> list = new ArrayList<>();
            //当前层的结点总数
            int count = queue.size();
            //遍历每一层
            while (count-- > 0 ) {
                //队头去除结点
                TreeNode node = queue.poll();
                //收集结果
                list.add(node.val);
                //左右节点都按照书匈奴添加到队尾
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            res.add(list);
        }
        return res;
    }
}