给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
首先让根节点入队列,然后遍历队列。
每次循环开始,计算出队列大小,当前的队列大小size就是每一层的节点数。循环这size个节点,依次弹出队列,然后判断当前节点的左右节点时候为空,非空的话入队。
迭代,直到队列为空。
/** * 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>> res = new LinkedList(); Queue<TreeNode> que = new LinkedList(); if(root != null){ que.add(root); } while(!que.isEmpty()){ int size = que.size(); List<Integer> arr = new LinkedList<>(); for(int i = 0; i < size; i++){ TreeNode temp = que.remove(); arr.add(temp.val); if(temp.left != null){ que.add(temp.left); } if(temp.right!=null){ que.add(temp.right); } } res.add(arr); } return res; } }