面试题 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]]
思路
看得出来是一层层的建立链表,可以很自然想到用队列层序遍历,每次在一层的循环中弹出元素,我们就根据元素的值创建链表的结点。但是链表头结点本身会参与向后的遍历,所以我们还要在一层的循环前创建一个虚拟头节点指向真正的头节点
源代码
class Solution {
public:
vector<ListNode*> listOfDepth(TreeNode* tree) {
vector<ListNode*> v;
if (tree == nullptr) {
return v;
}
queue<TreeNode*> queue;
queue.push(tree);
while (!queue.empty()) {
int size = queue.size();
ListNode* virtual_head = new ListNode();// 创建一个虚拟头节点用于指向真正的头节点
ListNode* head = virtual_head;// 创建头节点,后续会不断往后
for (int i=0; i<size; i++) {
TreeNode* node = queue.front();
queue.pop();
head->next = new ListNode(node->val);// 创建同值的链表结点,并让头节点指向它
head = head->next;// 头节点往后
if (node->left) {
queue.push(node->left);// 压入
}
if (node->right) {
queue.push(node->right);// 压入
}
}
v.push_back(virtual_head->next);// 每次创建好一层的链表,就把该链表的头结点推入vector
}
return v;
}
};