给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
思路:可以利用队列先进先出的特性,将将二叉树中的节点按层次放进队列中,然后再依次出队列。
具体代码如下:
/**
* 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>> ret = new ArrayList<>();
if(root==null) {
return ret;
}
Queue<TreeNode> q = new LinkedList<>();
q.offer(root); //先将根节点放入队列中
while(!q.isEmpty()) { //然后开始循环遍历
int size = q.size(); //size为当前层的元素个数
List<Integer> level = new ArrayList<>(size);//用level来保存一层元素
//一次性遍历一层的元素
for(int i=0;i<size;i++) {
TreeNode cur = q.poll(); //将当前层节点依次出队列
level.add(cur.val); //将当前层节点值放进level中
//如果节点有左孩子,就将左孩子入队列
if(cur.left != null) {
q.offer(cur.left);
}
//如果节点有右孩子,就将右孩子入队列
if(cur.right != null) {
q.offer(cur.right);
}
}
ret.add(level); //每遍历完一层就将当前层结果放入ret中
}
return ret;
}
}