题目描述:
标签:树 广度优先搜索 二叉树
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
代码:
思路分析:层序遍历的思想,不过需要每层添加一次。
/** * 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>> ans = new ArrayList<>(); if(root == null){ return ans; } Queue<TreeNode> queue = new LinkedList<>(); queue.add(root); while(!queue.isEmpty()){ int level = queue.size(); List<Integer> list = new ArrayList<>(); for(int i = 0;i < level;i++){ TreeNode node = queue.poll(); list.add(node.val); if(node.left != null){ queue.add(node.left); } if(node.right != null){ queue.add(node.right); } } ans.add(new ArrayList<>(list)); } return ans; } }