这篇文章主要为大家展示了“golang刷leetcode 技巧之如何实现特定深度节点链表”,内容简而易懂,条理清晰,希望能够帮助大家解决疑惑,下面让小编带领大家一起研究并学习一下“golang刷leetcode 技巧之如何实现特定深度节点链表”这篇文章吧。
给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 D,则会创建出 D 个链表)。返回一个包含所有深度的链表的数组。
示例:
输入:[1,2,3,4,5,null,7,8]
1
/ /
2 3
/ / /
4 5 7
/
8
输出:[[1],[2,3],[4,5,7],[8]]
解题思路
1,这是一道树的层序遍历+链表的组合题,主要考察树、队列、链表的理解、以及这几种数据结构的综合应用。
2,树的层序遍历,需要借助队列,对于每一层都分开需要借助两个队列
3,遍历的时候,我们将q1 这一层的每一个节点依次弹出,放入链表
4,将每一个节点的左右孩子依次存入q2
5,将q2存入q1
6,重复3~5,直到每个队列都为空为止。
代码实现
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func listOfDepth(tree *TreeNode) []*ListNode {
var q1,q2 queue
if tree==nil{
return nil
}
var r []*ListNode
q1.push(tree)
head:=new(ListNode)
for !q1.empty() || !q2.empty(){
cur:=head
for !q1.empty(){
x:=q1.pop()
if x.Left!=nil{
q2.push(x.Left)
}
if x.Right!=nil{
q2.push(x.Right)
}
cur.Next=&ListNode{
Val:x.Val,
}
cur=cur.Next
}
r=append(r,head.Next)
head.Next=nil
for !q2.empty(){
q1.push(q2.pop())
}
}
return r
}
type queue struct{
data []*TreeNode
}
func(this*queue)push(t*TreeNode){
this.data=append(this.data,t)
}
func (this*queue)empty()bool{
return len(this.data)<1
}
func (this*queue)pop()*TreeNode{
if this.empty(){
return nil
}
x:=this.data[0]
this.data=this.data[1:]
return x
}
以上是“golang刷leetcode 技巧之如何实现特定深度节点链表”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注亿速云行业资讯频道!
原创文章,作者:745907710,如若转载,请注明出处:https://blog.ytso.com/224015.html