力扣--面试题 04.03. 特定深度节点链表（中等题）

这题考察了二叉树的遍历和链表，我采用的是层序遍历，利用队列完成。

原题传送门

【题目描述】

给定一棵二叉树，设计一个算法，创建含有某一深度上所有节点的链表（比如，若一棵树的深度为 D，则会创建出 D 个链表）。返回一个包含所有深度的链表的数组。

【示例】

【解题过程】

【思路】

借用队列进行层序遍历，在每一层，把结点存入链表，把每个结点的左右子节点存入队列，然后删除该结点。

【代码】

/**
 * 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) {}
 * };
 */
class Solution {
public:
    vector<ListNode*> listOfDepth(TreeNode* tree) {
        vector<ListNode*> ans;
        if(tree == NULL){
            return ans;
        }
        TreeNode* tep=tree;
        queue<TreeNode*> que;
        que.push(tree);
        while(!que.empty()){//队列为空时结束
            int len=que.size();//获取本层的结点的个数
            ListNode* nodes=new ListNode(que.front()->val);
            ListNode* first=nodes;//第一个点
            if(que.front()->left!=NULL){
                que.push(que.front()->left);
            }
            if(que.front()->right!=NULL){
                que.push(que.front()->right);
            }
            que.pop();
            for(int i=1;i<len;i++){//遍历本层
                ListNode* ln=new ListNode(que.front()->val);//存入链表
                nodes->next=ln;
                nodes=nodes->next;
                if(que.front()->left!=NULL){//加入左右子节点
                que.push(que.front()->left);
            }
            if(que.front()->right!=NULL){
                que.push(que.front()->right);
            }
            que.pop();//删除此节点
            }
            ans.push_back(first);//把链表加入数组
        }
        return ans;
    }
};

运行结果：