BFS广度优先搜索技巧

剑指 Offer 32 - II. 从上到下打印二叉树 II

从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。

BFS广度优先搜索技巧_第1张图片

注意事项,queue中类型是Treenode,size每次遍历完一层要减1。
思路:每次遍历完一层添加进list,每一层中如果有左右子节点则加入队列。每一次遍历都让队列里的当前层出队,若是队列为空则结束。
答案:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
     
    public List<List<Integer>> levelOrder(TreeNode root) {
     
       
       List<List<Integer>> listAll = new LinkedList<>();
       Queue<TreeNode> queue = new ArrayDeque<>();
       if(root == null){
     
           return listAll;
       }
        queue.offer(root);
        //不为空结束
        while(!queue.isEmpty()){
     
        	//获得队列中元素的数量
            int size = queue.size();
            List<Integer> list = new LinkedList<>();
            //遍历当前层
            while(size>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);
                }
                //控制每一次遍历一层
                size--;
            }
            listAll.add(list);
        }

       return listAll;
    }
}

你可能感兴趣的:(算法与数据结构,队列,算法,二叉树)