輸出單層結點

• 對於一棵二叉樹，請設計一個算法，建立含有某一深度上全部結點的鏈表。 給定二叉樹的根結點指針TreeNode* root，以及鏈表上結點的深度，請返回一個鏈表ListNode，表明該深度上全部結點的值，請遵循樹上從左往右的順序連接，保證深度不超過樹的高度，樹上結點的值爲非負整數且不超過100000。
• 答案：這個題目的意思就是輸出二叉樹的某一層的全部元素，這個首先想到的是層次遍歷，層次遍歷最簡單的方法就是用隊列實現，咱們傳統的層次遍歷方法是能夠輸出全部元素，那麼如何區分相鄰兩層之間的元素呢？ 其實咱們能夠用兩個整數變量line1，line2來記錄相鄰兩層的元素個數，其中line1表明出棧那一層留下的元素個數，line2表明下一層進棧元素的個數，每當line1爲0的時候，說明上一層已經所有出棧，下一層已經所有入棧，那麼層次遍歷層數就加一，這個時候將line2的值複製給line1，line2=0，當遍歷到第dep層的時候，便把那一層的全部元素輸出，中止遍歷。代碼實現以下：

/*
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) {
        // write code here
        if(dep<=0||root==NULL)
            return NULL;
        ListNode* list=new ListNode(-1);
        ListNode* listHead=list;
        queue<TreeNode*> qu;
        qu.push(root);
        int lines1=1,lines2=0,num=1;
        while(!qu.empty())
        {
            if(num==dep)
            {
                for(int i=0;i<lines1;i++)
                {
                    TreeNode* root1=qu.front();
                    list->next=new ListNode(root1->val);
                    list=list->next;
                    qu.pop();
                }
                return listHead->next;
            }
            TreeNode* root1=qu.front();
            if(root1->left)
            {
                qu.push(root1->left);
                lines2++;
            }
            if(root1->right)
            {
                qu.push(root1->right);
                lines2++;
            }
            qu.pop();
            if(--lines1==0)
            {
                lines1=lines2;
                lines2=0;
                num++;
            }
        }
        return listHead->next;
    }
};