面试题：特定深度结点链表

一.问题描述

给定一棵二叉树，设计一个算法，创建含有某一深度上所有节点的链表（比如，若一棵树的深度为 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）
链接：题目来源

二.问题分析

1）这道题实质上还是进行二叉树的层序遍历，只是与上一次做的题（上次题目）不一样，它要求我们使用链表来存放每一层的元素。
2）最容易的做法还是广度优先搜素，以队列作为数据结构，以链表结点作为队列类型。
3)注意在指向下一个结点的时候需要new一块空间。
4）二叉树和链表的说明：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

5）见如下代码

三.代码

class Solution {
public:
    vector<ListNode*> listOfDepth(TreeNode* tree) {
        vector<ListNode*> res;  //目标数组
        if (tree == NULL) return res;  //判空
        queue<TreeNode*> q; //队列声明
        q.push(tree);  //将根结点的指针放入队列 
        while(!q.empty()) //循环条件，队列不为空
        {
            int n = q.size();  //循环次数，即该层的元素个数
            ListNode temp(0),*ans=nullptr;  //temp为链表结点的定义，初始化val值为0，ans为指向它的指针
            ans = &temp;
            for(int i=0;i<n;i++)  
            {
                TreeNode* root = q.front(); //找到该层第一个元素，并出队
                q.pop();
                ans = ans->next = new ListNode(root->val);  //开辟一个新的空间使ans指针指向这个新值
                if(root->left) q.push(root->left);  //判断是否有左孩子，若有则入队
                if(root->right) q.push(root->right); //判断是否有右孩子，若有则入队
            }
            //因为temp的值为空，temp的next才是第一个值的存放地址，所以将temp.next放入目标数组
            res.emplace_back(temp.next); 
        }
        return res;
    }
};