输出树单层结点

题目描述

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

class TreeLevel {
public:
    ListNode* getTreeLevel(TreeNode* root, int dep) {
        // write code here
        if(root == NULL)
            return NULL;
        //init
        int count = -1;
        queue<TreeNode*> que;
        TreeNode* level = new TreeNode(count);
        //第一层,序号在结点之前
        que.push(level);
        que.push(root);
        count--;
        //第二层
        while (!que.empty()){
            TreeNode* tmp = que.front();
            que.pop();
            int key = tmp->val;
            //结点
            if(key >= 0){
                if(tmp->left != NULL)
                    que.push(tmp->left);
                if(tmp->right != NULL)
                    que.push(tmp->right);
            }
                //层标志
            else{
                //到达所要的层
                if(dep == -key){
                    ListNode* head = new ListNode(0);
                    ListNode* p = head;
                    while (!que.empty()){
                        //树操作，拿出队列的后续所有数据
                        key = que.front()->val;
                        que.pop();
                        //链表操作
                        ListNode* list_tmp = new ListNode(key);
                        p->next = list_tmp;
                        p = p->next;
                    }
                    return head->next;
                }
                    //还没到指定层
                else{
                    tmp->val = count;
                    count--;
                    que.push(tmp);
                }
            }
        }
        return NULL;
    }
};