面试题 04.03. 特定深度节点链表

最新推荐文章于 2021-06-18 00:07:19 发布

vision泉

最新推荐文章于 2021-06-18 00:07:19 发布

阅读量184

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_38603360/article/details/105443244

版权

本文介绍了一种通过层次遍历二叉树的方法，将同一深度的节点转换为链表结构，实现特定深度节点链表的功能。使用了队列进行层次遍历，将每一层的节点值插入到链表中，最终返回所有深度的链表集合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

面试题 04.03. 特定深度节点链表

思路：层次遍历

/**
 * 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*> res;
        if(tree==NULL) return res;
        queue<TreeNode*> q;
        q.push(tree);
        while(!q.empty()){
            int size = q.size();
            ListNode* tmp = new ListNode(0);
            ListNode* Lnow = tmp;
            for(int i=0;i<size;i++){
                TreeNode* now = q.front();
                Lnow->next = new ListNode(now->val);
                Lnow = Lnow->next;
                q.pop();
                if(now->left) q.push(now->left);
                if(now->right) q.push(now->right);
            }
            if(tmp->next)
                res.push_back(tmp->next);
        }
        return res;
    }
};