讨论/《零起步学算法》 - 练习:二叉树的层次遍历 II/
《零起步学算法》 - 练习:二叉树的层次遍历 II
共 1 个回复

推荐使用offer,不要使用add

import java.util.LinkedList;
 import java.util.List;
 import java.util.ArrayList;
 import java.util.Queue;
class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
      //广度优先搜索 => 层级遍历
      //创建链表
      List<List<Integer>> res = new LinkedList<>();
      //判空
      if(root == null) {
        return  res;
      }
      //创建队列,用来存树节点
      Queue<TreeNode> queue = new LinkedList<>();
      //根结点放入队列
      queue.add(root);
      //遍历
      while(!queue.isEmpty()) {
        List<Integer> curLevel = new ArrayList<>();
        //出队列
        int curSize = queue.size();
        for(int i = 0; i < curSize; i++) {
          TreeNode head = queue.poll();
          curLevel.add(head.val);
          if(head.left != null) {
            queue.offer(head.left);
          }
          if(head.right != null) {
            queue.offer(head.right);
          }
        }
        res.add(0,curLevel);
      }
      return res;
    }
}
2