面试题 04.03. 特定深度节点链表
给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 D,则会创建出 D 个链表)。返回一个包含所有深度的链表的数组。
示例:
输入:[1,2,3,4,5,null,7,8]
1
/ \
2 3
/ \ \
4 5 7
/
8
输出:[[1],[2,3],[4,5,7],[8]]
思路
1. 题意:二叉树层次遍历,每层以链表形式连接。
2. 方法:两部分构成:层序遍历和链表连接。
层序遍历:使用队列存储树节点,每次移除该节点,并向队列添加该节点对应的叶子节点。
链表连接:向链表中添加从队列中移除的节点。
代码中的subResult表示每层树节点的链表(赋予初值0),current在该链表上移动。向结果集添加每层链表时,添加subResult.next用于跳过初值。
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode[] listOfDepth(TreeNode tree) {
ArrayList<ListNode> result = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(tree);
while (!queue.isEmpty()) {
ListNode subResult = new ListNode(0);
int size = queue.size();
ListNode current = subResult;
for (int i = 0; i < size; i++) {
TreeNode treeNode = queue.poll();
current.next = new ListNode(treeNode.val);
if (treeNode.left != null)
queue.offer(treeNode.left);
if (treeNode.right != null)
queue.offer(treeNode.right);
current = current.next;
}
result.add(subResult.next);
}
return result.toArray(new ListNode[0]);
}
}