[英]Level order traversal binary tree using two queues
Leetcode 问题: https ://leetcode.com/problems/binary-tree-level-order-traversal/
我有两个队列,我清空我的第一个队列 (q),同时将元素添加到我的第二个队列 (q2),这样我就可以获得级别。 当第一个队列为空时,我将它传递给函数以创建该级别的 ArrayList 并添加到我的结果值中,如果 q2 为空,则意味着没有添加任何内容,因此我们可以跳出循环并返回该函数. 我必须手动添加第一个节点。 问题是只添加了中间级别,而不是其他级别。
/**
* 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) {
if(root == null) {
return null;
}
List<List<Integer>> returnList = new ArrayList<>();
Queue<TreeNode> q = new LinkedList<>();
q.add(root);
List<Integer> tempList = new ArrayList<>();
tempList.add(root.val);
returnList.add(tempList);
while(true) {
Queue<TreeNode> q2 = new LinkedList<>();
while(!q.isEmpty()) {
TreeNode node = q.remove();
if(node.left != null) {
q2.add(node.left);
}
if(node.right != null) {
q2.add(node.right);
}
}
if(q2.isEmpty()) {
break;
}
q = q2;
addList(returnList, q2);
}
return returnList;
}
public static void addList(List<List<Integer>> returnList, Queue<TreeNode> q2) {
List<Integer> tempList = new ArrayList<>();
int size = q2.size();
for(int i = 0; i < size; i++) {
int val = q2.remove().val;
System.out.println(val);
tempList.add(val);
}
returnList.add(tempList);
}
}
问题与这部分代码有关:
q = q2;
addList(returnList, q2);
函数addList
将清空给定的队列q2
,对其调用remove
。 由于q
已成为对同一队列的引用,您已清空q
,因此外循环的下一次迭代将不会进入内循环,并以break
退出外循环。
为了避免这种情况发生,请确保没有在q
执行remove
调用:制作一个副本:
q = new LinkedList<>(q2);
addList(returnList, q2);
或者,或者:
q = q2;
addList(returnList, new LinkedList<>(q2));
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.