层序遍历在今天的应用有求二叉树或N叉树的最大或最小深度以及节点个数。
首先就是二叉树和N叉树的区别:二叉树是只保证有两个子节点,一个left和一个right;而N叉树的子节点就是大于两个了,表示为children子数组。因此,在求二叉树或N叉树的最大或最小深度以及节点个数时,使用到的方法一模一样,只要在操作子元素时修改一下即可。最大/小深度的定义:从根节点到距离最远的叶子节点之间的节点个数或者边的条数,具体应以题目为主。
以求二叉树和N叉树的最大深度为例(根节点到距离最远的叶子节点之间的节点个数):
//二叉树最大深度
var maxDepth = function (root, d = 0) {
if (root) {
let queue = [root]
while (queue.length) {
let size = queue.length
while (size--) {
let a = queue.shift()
if (a.left) queue.push(a.left)//判断左右节点是否为空,不为空则加入队列
if (a.right) queue.push(a.right)
}
d++
}
}
return d
};
//N叉树最大深度
var maxDepth = function(root,d=0) {
if(root){
let queue=[root]
while(queue.length){
let size = queue.length
while(size--){
let a = queue.shift()
if(a.children.length){//判断是否有子节点,有的话就将所有子节点加入队列。
for(let i = 0;i<a.children.length;i++){
queue.push(a.children[i])
}
}
}
d++
}
}
return d
};
最小深度思路:最小深度就是从根节点到距离最近的叶子节点之间的边的条数或者节点个数,只需在上述代码判断每个子节点是否有子节点,没有的话该层就是最小深度。
节点个数思路:这个没什么,只需使用任意一种遍历方式,得到所有节点的数量即可。