题目链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
经典BFS与简单链表结合的题目。
#define MAX_DEPTH (1000)
struct ListNode** listOfDepth(struct TreeNode* tree, int* returnSize)
{
*returnSize = 0;
struct ListNode **ans = (struct ListNode **)malloc(sizeof(struct ListNode*) * MAX_DEPTH);
struct TreeNode *quene[MAX_DEPTH];
struct ListNode *pre = NULL;
int front = 0;
int rear = 0;
if (tree) {
quene[rear++] = tree;
}
while (front != rear) {
int cnt = rear - front;
for (int i = 0; i < cnt; i++) {
if (i == 0) {
ans[(*returnSize)] = (struct ListNode*)malloc(sizeof(struct ListNode)); // 每层的链表头
pre = ans[(*returnSize)++];
pre->val = quene[front]->val;
pre->next = NULL;
} else {
struct ListNode *node = (struct ListNode*)malloc(sizeof(struct ListNode));
node->val = quene[front]->val;
node->next = NULL;
pre->next = node;
pre = pre->next;
}
if (quene[front]->left) {
quene[rear++] = quene[front]->left;
}
if (quene[front]->right) {
quene[rear++] = quene[front]->right;
}
front++;
}
}
return ans;
}