前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >程序员面试金典 - 面试题 04.03. 特定深度节点链表(BFS)

程序员面试金典 - 面试题 04.03. 特定深度节点链表(BFS)

作者头像
Michael阿明
发布2020-07-13 15:50:20
2420
发布2020-07-13 15:50:20
举报

1. 题目

给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 D,则会创建出 D 个链表)。返回一个包含所有深度的链表的数组。

代码语言:javascript
复制
例:
输入:[1,2,3,4,5,null,7,8]

        1
       /  \ 
      2    3
     / \    \ 
    4   5    7
   /
  8

输出:[[1],[2,3],[4,5,7],[8]]

2. 解题

  • 层序,BFS遍历
代码语言:javascript
复制
class Solution {
public:
    vector<ListNode*> listOfDepth(TreeNode* tree) {
    	if(!tree)
    		return {};
    	queue<TreeNode*> q;
    	vector<ListNode*> ans;
    	q.push(tree);
    	int size;
    	TreeNode* tp;
    	ListNode* head, *cur;
    	bool hashead;
    	while(!q.empty())
    	{
    		size = q.size();
    		hashead = false;
    		while(size--)
    		{
    			tp = q.front();
    			q.pop();
    			if(!hashead)
    			{
    				head = new ListNode(tp->val);
    				cur = head;
    				hashead = true;
    			}
    			else
    			{
    				cur->next = new ListNode(tp->val);
    				cur = cur->next;
    			}
    			if(tp->left)
    				q.push(tp->left);
    			if(tp->right)
    				q.push(tp->right);
    		}
    		ans.push_back(head);
    	}
    	return ans;
    }
};
在这里插入图片描述
在这里插入图片描述
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-03-20 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档