/**
* 主要思想:主要思想就是使用两个队列一个队列指向上一层,另外一个队列则指向下一层,上层的队列
* 出队时判断其是否存在子节点有则存入下层队列中。当一层遍历结束后下层的队列也就形成了,这时将
* 这时的处理就是将将上层队列下移一层,下层的队列清空重新构建更下层的队列。如此循环往复,直到
* 下层不再有子节点,这时两个队列都是空队列遍历结束
* */
public ArrayList<ArrayList> levelOrder (TreeNode root) {
ArrayList<ArrayList> result = new 、 ArrayList<ArrayList>();
if(root == null){
return result;
}
ArrayDeque queue_1 = new ArrayDeque();
ArrayDeque queue_2 = new ArrayDeque();
queue_1.offer(root);
int level = 0;
ArrayList list = new ArrayList();
while(queue_1.size() != 0 || queue_2.size() != 0) {
TreeNode node = queue_1.pop();
list.add(node.val);
if(node.left != null) {
queue_2.offer(node.left);
}
if(node.right != null) {
queue_2.offer(node.right);
}
if(queue_1.size() == 0) {
queue_1 = queue_2;
queue_2 = new ArrayDeque();
result.add(list);
list = new ArrayList();
}
}
return result;
}
使用单层循环完成树的层次遍历
最新推荐文章于 2022-04-23 18:37:16 发布