[LeetCode]102. 二叉树的层序遍历(java实现)
1. 题目
2. 读题(需要重点注意的东西)
思路(层次遍历):
将根节点进队
,当队中不为空时,将队首元素出队,然后将队首元素的左子树和右子树入队,直到队空;
如何判断节点属于哪一层?
在每次出队的时候,用count
记录此时此刻队列中的元素,这些元素都属于同一层,每当出队一个元素,count--
,最后将这一层的元素加到结果链表中即可。
3. 解法
---------------------------------------------------解法---------------------------------------------------:
/**
* 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 new ArrayList<>();
Queue<TreeNode> q = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
q.offer(root);
while(!q.isEmpty()){
int count = q.size();
List<Integer> list = new ArrayList<>();
while(count > 0){
TreeNode temp = q.poll();
list.add(temp.val);
if(temp.left != null) q.offer(temp.left);
if(temp.right != null) q.offer(temp.right);
count--;
}
res.add(list);
}
return res;
}
}
可能存在的问题: