题目描述:
思路:二叉树的层序遍历并不难,难的是如何判定分层。官方是这么做的:
但我用了两个队列实现的这个做法,一个节点出队列A后,将其孩子结点入队列B,待A空队列后B队列元素给A。其中有几点需要注意。1.队列B元素给A不能单纯的q1=q2,q2.clear()。这样会使队列1也成为空队列。2. list=new LinkedList<>();不能换成list.clear(),这样会使得已加入result的list变空,原因是一直用的都是list这一个链表,而不是新的链表。
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
class Solution {
public static void main(String[] args) {
TreeNode n1=new TreeNode(3);
TreeNode n2=new TreeNode(9);
TreeNode n3=new TreeNode(20);
TreeNode n4=new TreeNode(15);
TreeNode n5=new TreeNode(7);
n1.left=n2;
n1.right=n3;
n3.left=n4;
n3.right=n5;
new Solution().levelOrder(n1);
}
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> result = new LinkedList<>();
if (root == null) return result;
Queue<TreeNode> q1 = new LinkedList<>();
Queue<TreeNode> q2 = new LinkedList<>();
q1.add(root);
TreeNode mid = null;
List<Integer> list = new LinkedList<>();
while (!q1.isEmpty()) {
mid=q1.remove();
if (mid.left!=null)q2.add(mid.left);
if (mid.right!=null)q2.add(mid.right);
list.add(mid.val);
if (q1.isEmpty()){
while (!q2.isEmpty()){
q1.add(q2.remove());
}
result.add(list);
list=new LinkedList<>();
}
}
return result;
}
}