难度:中等 频率:150 题目: 给你一个二叉树,请你返回其层序遍历得到的节点值(逐层,从左到右访问所有节点)
做这道题 之前先回忆一下数据结构里的 BFS(Breath First Search,广度优先搜索)和DFS(Depth First Search)如何实现遍历。
void dfs(TreeNode root)
{
if(root!=null)
dfs(root.left);
dfs(root.right);
}
void bfs(TreeNode root)
{
Queue<TreeNode> queue=new ArrayDeque<>();
queue.add(root);
while(!queue.isEmpty()){
TreeNode node = queue.poll();
if(node.left!=null)
{
queue.add(node.left);
}
if(node.right!=null)
{
queue.add(node.right);
}
}
}
总结:BFS用来解决层级遍历问题,而DFS用来解决最短路径问题。
解题方法:BFS变形 解题思路: 1.在常规的BFS遍历中【队列】,加入一个参数,记录这一层的节点数量。 比如一开始root进入的时候,记录数量1。判断一左一右为一次。计次1次遍历,即遍历一棵小树。 然后进入2个,这个时候就会记录2。然后倒计时遍历2次,即遍历两个小树,或者理解为2个root。 2.每一次倒计时用一个arraylist,每次队列出去的数,都放在arraylist里。 3.最后再用一个大的arraylist,把之前每一次的arraylist装起来。
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<>();
Queue<TreeNode>queue=new ArrayDeque();
if(root!=null)
queue.add(root);
while(!queue.isEmpty()){
int n=queue.size();
List<Integer> arr = new ArrayList();
for(int i=0;i<n;i++)
{
TreeNode node =queue.poll();
arr.add(node.val);
if(node.left!=null)
{
queue.add(node.left);
}
if(node.right!=null)
{
queue.add(node.right);
}
}
res.add(arr);
} return res;
}
}
PS:易错点
- 双向队列里面装的是树节点,不是普通类型。
- 一开始的根节点需要判断是否为空。
- list添加的都是值,不是树节点。
- 队列的里节点的数量 是 queue.size()。
- list里套list。二位数组。
|