用c++编程,给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 D,则会创建出 D 个链表)。返回一个包含所有深度的链表的数组。 要二叉树的深度指的是这棵树有几层,比如下面的例子深度为4,也就是需要返回1个包含4个链表的数组,并逐句解释代码
时间: 2024-03-24 14:40:59 浏览: 61
好的,我会尽力回答你的问题。以下是解决这个问题的C++代码及其解释:
```cpp
#include <vector>
#include <queue>
using namespace std;
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
vector<ListNode*> createLevelLinkedLists(TreeNode* root) {
vector<ListNode*> result; // 用于存储所有深度的链表的数组
if (!root) {
return result;
}
queue<TreeNode*> q;
q.push(root);
while (!q.empty()) {
int levelSize = q.size(); // 每一层节点的数量
ListNode* dummy = new ListNode(0);
ListNode* current = dummy;
for (int i = 0; i < levelSize; i++) {
TreeNode* node = q.front();
q.pop();
current->next = new ListNode(node->val);
current = current->next;
if (node->left) {
q.push(node->left);
}
if (node->right) {
q.push(node->right);
}
}
result.push_back(dummy->next);
}
return result;
}
```
这段代码的主要思路是通过广度优先搜索(BFS)来遍历二叉树,对于每一层来说,将其节点的值添加到一个链表中,最后将链表存储到一个vector中,这个vector就是题目要求的包含所有深度的链表的数组。
具体来说,我们首先判断根节点是否为空,如果是,则直接返回一个空的vector。否则,我们创建一个队列,并把根节点加入队列中。接下来,我们进入一个while循环,直到队列为空。在每一次循环中,我们首先获取当前队列的长度,这个长度就是当前层的节点数。为了将当前层的所有节点的值添加到一个链表中,我们创建一个dummy节点作为头结点,一个current指针指向dummy节点,然后遍历这一层的所有节点,将它们的值添加到链表中。同时,如果当前节点有左子节点或右子节点,都将它们加入队列中以便后续遍历。当遍历完一层之后,我们将dummy节点的下一个节点(即链表的实际头结点)添加到result数组中。最后,我们返回result数组即可。
希望这个解答能够帮助到你,如果你还有其他问题,请随时问我。
阅读全文