题目描述
给定一个 N 叉树,返回其节点值的_层序遍历_。(即从左到右,逐层遍历)。
树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。
示例 1:
输入: root = [1,null,3,2,4,null,5,6]
输出: [[1],[3,2,4],[5,6]]
示例 2:
输入: root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出: [[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]
提示:
- 树的高度不会超过
1000
- 树的节点总数在
[0, 10^4]
之间
代码
bfs
class Solution {
public List<List<Integer>> levelOrder(Node root) {
Queue<Node> queue = new LinkedList<>();
ArrayList<List<Integer>> lists = new ArrayList<>();
if (root == null) return lists;
lists.add(new ArrayList<Integer>(){{add(root.val);}});
for (Node child : root.children) {
queue.add(child);
}
while (!queue.isEmpty()) {
int len = queue.size();
ArrayList<Integer> temp = new ArrayList<>();
for (int i = 0; i < len; i++) {
Node poll = queue.poll();
temp.add(poll.val);
for (Node child : poll.children) {
queue.add(child);
}
}
lists.add(temp);
}
return lists;
}
}
dfs
class Solution {
List<List<Integer>> ans = new ArrayList<>();
public List<List<Integer>> levelOrder(Node root) {
dfs(root, 1);
return ans;
}
private void dfs(Node node, int deep) {
if (node == null) return;
if (ans.size() < deep) {
ans.add(new ArrayList<>());
}
ans.get(deep - 1).add(node.val);
for (Node child : node.children) {
dfs(child, deep + 1);
}
}
}
题解
bfs
借助队列来完成bfs操作
先按顺序遍历一层的所有节点, 并把他们的所有子节点按顺序放入队列中
如此一轮遍历后, 队列就被存满了下一层的的节点存满了
如法炮制的多次循环把直到队列为空就得到答案了
dfs
用函数传递的参数表示当前栈的深度(从1开始)
应为深度也可以对应List下标, 所以很容易对对当前栈代表节点的值进行存储
具体代码参上
好的!本次分享到这就结束了
如果对铁汁你有帮助的话,记得点赞👍+收藏⭐️+关注➕
我在这先行拜谢了:)