解题思路:
创建一个队列(利用队列先进先出的特征)
将二叉树的头节点加入队列
队列删除第一个元素,定义一个节点变量temp,用来存储队列中出来的元素,利用temp找到第一个出来的元素的左右节点,并存入队列。如此循环,直到队列中的元素为0 ,也就是二叉树的元素全部存进列表中
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
/**
*
* @param root TreeNode类
* @return int整型ArrayList<ArrayList<>>
*/
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
// write code here
ArrayList<ArrayList<Integer>> arr = new ArrayList<ArrayList<Integer>>();
//定义一个队列,用于存放当前层数的元素
Queue <TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
if(root == null){
return arr;
}
while(!queue.isEmpty()){
ArrayList <Integer> sin = new ArrayList<>();
int size = queue.size();//得到队列长度,获取循环的次数
for(int i=0; i<size; i++){
TreeNode temp = queue.poll();
sin.add(temp.val);
if(temp.left != null){
queue.offer(temp.left);
}
if(temp.right != null){
queue.offer(temp.right);
}
}
arr.add(sin);
}
return arr;
}
}