107. 二叉树的层序遍历 II

image.png

层序遍历

  1. public class Solution {
  2.     // 二叉树层序遍历使用广度优先搜索
  3.     // 题目要求反向遍历，所以将每一层的遍历结果放到 res 的头部即可
  4.     public List<List<Integer>> levelOrderBottom(TreeNode root) {
  5.         // 用于存放遍历结果
  6.         List<List<Integer>> res = new ArrayList<>();
  7.         // 广度搜索辅助队列
  8.         Queue<TreeNode> queue = new LinkedList<>();
  9.         if (root == null) return res;
  10.         queue.offer(root);
  12.         while (!queue.isEmpty()) {
  13.             // 用于存放当前层的遍历结果
  14.             List<Integer> listTemp = new ArrayList<>();
  15.             int size = queue.size();
  16.             for (int i = 0; i < size; i++) {
  17.                 // 从左往右取出节点
  18.                 TreeNode cur = queue.poll();
  19.                 listTemp.add(cur.val);
  21.                 // 分别将左右树进队
  22.                 if (cur.left != null) queue.add(cur.left);
  23.                 if (cur.right != null) queue.add(cur.right);
  24.             }
  25.             // 放入头部，因为题目要求反向遍历
  26.             res.add(0, listTemp);
  27.         }
  28.         return res;
  29.     }
  30. }