给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 D,则会创建出 D 个链表)。返回一个包含所有深度的链表的数组。
示例:
输入:[1,2,3,4,5,null,7,8]
1
/ \
2 3
/ \ \
4 5 7
/
8
输出:[[1],[2,3],[4,5,7],[8]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/list-of-depth-lcci
思路分析:二叉树层次遍历
AC代码
public ListNode[] listOfDepth(TreeNode tree) {
if (tree == null) {
return new ListNode[0];
}
List<ListNode> lists = new ArrayList<>(10);
// 队列,先进先出
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(tree);
while (!queue.isEmpty()) {
ListNode list = null;
// 算出当前队列的大小,即上一层节点的数量
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
// 把上一层节点放入链表list
if (list == null) {
list = new ListNode(node.val);
lists.add(list);
} else {
list.next = new ListNode(node.val);
list = list.next;
}
// 把左右节点放入队列
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
}
ListNode[] arr = new ListNode[lists.size()];
for (int i = 0; i < lists.size(); i++) {
arr[i] = lists.get(i);
}
return arr;
}