二叉树的深度
function TreeNode(val) {
this.val = val;
this.left = this.right = null;
}
let tree = {
val: 3,
left: {
val: 9
},
right: {
val: 20,
left: {
val: 15
},
right: {
val: 7
}
}
}
方法一:后序遍历(DFS)
树的后序遍历 / 深度优先搜索往往利用 递归 或 栈 实现,本文使用递归实现。
关键点: 此树的深度和其左(右)子树的深度之间的关系。显然,此树的深度 等于 左子树的深度 与 右子树的深度 中的 最大值 +1 。
var maxDepth = function (root) {
if (root == null) {
return 0;
}
return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
};
算法解析:
终止条件: 当 root 为空,说明已越过叶节点,因此返回 深度 0。
递推工作: 本质上是对树做后序遍历。
计算节点 root 的 左子树的深度 ,即调用 maxDepth(root.left)
;
计算节点 root 的 右子树的深度 ,即调用 maxDepth(root.right)
;
返回值: 返回 此树的深度 ,即 max(maxDepth(root.left), maxDepth(root.right)) + 1
。
方法二:层序遍历(BFS)
树的层序遍历 / 广度优先搜索往往利用 队列 实现。
关键点: 每遍历一层,则计数器 +1,直到遍历完成,则可得到树的深度。
var maxDepth = function (root) {
if (root == null) {
return 0;
}
let queue = [];
queue.unshift(root);
let tmp = [];
let res = 0;
while (queue.length !== 0) {
tmp = [];
for (node of queue) {
if (node.left != null) {
tmp.unshift(node.left);
}
if (node.right != null) {
tmp.unshift(node.right);
}
}
queue = tmp;
res++;
}
return res;
};
let length = maxDepth(tree);
console.log(length);
算法解析:
特例处理: 当 root 为空,直接返回 深度 0 。
初始化: 队列 queue (加入根节点 root ),计数器 res = 0。
循环遍历: 当 queue 为空时跳出。
初始化一个空列表 tmp ,用于临时存储下一层节点;
遍历队列: 遍历 queue 中的各节点 node ,并将其左子节点和右子节点加入 tmp;
更新队列: 执行 queue = tmp ,将下一层节点赋值给 queue;
统计层数: 执行 res += 1 ,代表层数加 1;
返回值: 返回 res 即可。