面试算法-----输出单层结点

题目描述
对于一棵二叉树，请设计一个算法，创建含有某一深度上所有结点的链表。

给定二叉树的根结点指针TreeNode* root，以及链表上结点的深度，请返回一个链表ListNode，代表该深度上所有结点的值，请按树上从左往右的顺序链接，保证深度不超过树的高度，树上结点的值为非负整数且不超过100000。

直接用BFS就可以解决了。搜索到该层时候，把同一层的结点保存在一个链表中，直接返回该链表。

/* 
struct TreeNode { 
    int val; 
    struct TreeNode *left; 
    struct TreeNode *right; 
    TreeNode(int x) : 
            val(x), left(NULL), right(NULL) { 
    } 
};*/  
  
/* 
struct ListNode { 
    int val; 
    struct ListNode *next; 
    ListNode(int x) : val(x), next(NULL) {} 
};*/  
class TreeLevel  
{  
    public:  
        ListNode* getTreeLevel(TreeNode* root, int dep)  
        {   
            if(root==nullptr)  
                return nullptr;  
            queue<TreeNode*> Q,tmp;  
            Q.push(root);  
            int cnt = 1;  
            ListNode *newList = new ListNode(0);  
            ListNode *pNode = newList;  
            while(!Q.empty())  
            {  
                if(cnt==dep)  
                {  
                    while(!Q.empty())  
                    {  
                        ListNode *newNode = new ListNode(Q.front()->val);  
                        Q.pop();  
                        pNode->next = newNode;  
                        pNode = pNode->next;  
                    }  
                }  
                while(!Q.empty())  
                {  
                    TreeNode *cur = Q.front();  
                    Q.pop();  
                    tmp.push(cur->left);  
                    tmp.push(cur->right);  
                }  
                while(!tmp.empty())  
                {  
                    Q.push(tmp.front());  
                    tmp.pop();  
                }  
                ++cnt;  
            }  
            return newList->next;  
        }  
};