面试题 04.03. 特定深度节点链表(层次遍历)

最新推荐文章于 2022-09-01 22:06:08 发布

Golden age

最新推荐文章于 2022-09-01 22:06:08 发布

阅读量113

点赞数

分类专栏： C/C++解Leetcode 文章标签：二叉树 leetcode

本文链接：https://blog.csdn.net/qq_40735907/article/details/113619404

版权

C/C++解Leetcode 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

在这里插入图片描述

class Solution {
public:
    vector<ListNode*> listOfDepth(TreeNode* tree) {
        queue<TreeNode*> q;
        q.push(tree);
        vector<ListNode*> ret;
        while(!q.empty()){
            int size = q.size();
            ListNode* head = new ListNode(0);
            ListNode* p = head;
            while(size--){
                TreeNode* cur = q.front();
                q.pop();
                if(cur->left){
                    q.push(cur->left);
                }
                if(cur->right){
                    q.push(cur->right);
                }
                p->next = new ListNode(cur->val);
                p = p->next;
            }
            ret.push_back(head->next);
            delete head;
        }
        return ret;
    }
};