LeetCode-102-二叉树的层序遍历

题目描述:
给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。

题目链接:LeetCode-102-二叉树的层序遍历
LeetCode-102-二叉树的层序遍历_第1张图片

解题思路:
使用队列 先进先出的特点存储每次遍历的节点。

代码实现:

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        bfs(root);
        return res;
    }

	// 层遍历
    private void bfs(TreeNode root){
        if (root==null){
            return;
        }

        // 使用 队列 来遍历并保存,每次记录队列的大小size,
        // 每次弹出一个,在队尾加入该节点的左右孩子,同时size--
        // 当 size=0,说明这一层就结束了,更新size,更新后的 size大小就是下一层节点的个数
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);// 根节点入队列
        while (!queue.isEmpty()) {
            List<Integer> path = new ArrayList<>();// 每更新遍历一层的时候都要新建一个
            int size = queue.size();// 更新 size
            while (size>0){
                // 弹出一个
                TreeNode tmp = queue.poll();
                path.add(tmp.val);// path 记录队列弹出的元素
                // 左孩子入队列
                if (tmp.left != null) {
                    queue.add(tmp.left);
                }
                // 右孩子入队列
                if (tmp.right != null) {
                    queue.add(tmp.right);
                }
                size--;
            }
            // 出while循环说明这一层更新完了
            res.add(path);// 收获结果,不需要新建一个ArrayList<>(),因此每次的都是最新的
        }
    }
}

你可能感兴趣的:(LeetCode,leetcode,算法)