数据结构与算法之层次遍历算法

层次遍历算法

题目

原题：102. 二叉树的层序遍历 - 力扣（LeetCode）

给你二叉树的根节点root，返回其节点值的层序遍历。（即逐层地，从左到右访问所有节点）。

逻辑思路

• 建立一个queue
• 先把根节点放进去，这时候找根节点的左右两个子节点
• 去掉根节点，此时queue里的元素就是下一层的所有节点
• 用for循环遍历，将结果存到一个一维向量里
• 遍历完之后再把这个一维向量存到二维向量里
• 以此类推，可以完成层序遍历
  

代码

这个题目其实就是考察BFS求最短路径问题，🤣写这篇文章的时候还不会这招。见谅啦

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> lists = new ArrayList<>();
        if (root == null) {
            return lists;
        }
        Queue<TreeNode> treeNodeQueue = new ArrayDeque<>();
        treeNodeQueue.add(root);
        List<TreeNode> sameStorey = new ArrayList<>();
        while (!treeNodeQueue.isEmpty()) {
            //每当队列中弹出一个元素时，就需要将弹出元素子元素(左右)进行入队操作
            List<Integer> temp = new ArrayList<>();
            while (!treeNodeQueue.isEmpty()) {
                TreeNode poll = treeNodeQueue.poll();
                temp.add(poll.val);
                if (poll.left != null) {
                    sameStorey.add(poll.left);
                }
                if (poll.right != null) {
                    sameStorey.add(poll.right);
                }
            }
            //将子节点加入到队列中，并清空集合
            treeNodeQueue.addAll(sameStorey);
            sameStorey.clear();
            lists.add(temp);
        }
        return lists;
    }
}