PS:《剑指offer》是很多同学找工作都会参考的一本面试指南,同时也是一本算法指南(为什么它这么受欢迎,主要应该是其提供了一个循序渐进的优化解法,这点我觉得十分友好)。现在很多互联网的算法面试题基本上可以在这里找到影子,为了以后方便参考与回顾,现将书中例题用Java实现(第二版),欢迎各位同学一起交流进步。
GitHub: https://github.com/Uplpw/SwordOffer。
剑指offer完整题目链接: https://blog.csdn.net/qq_41866626/article/details/120415258
1 题目描述
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
leetcode链接: 二叉树的深度(以下代码已测试,提交通过)
2 测试用例
一般是考虑功能用例,特殊(边缘)用例或者是反例,无效测试用例这三种情况。甚至可以从测试用例寻找一些规律解决问题,同时也可以让我们的程序更加完整鲁棒。
(1)功能用例:完全与非完全二叉树。
(2)边缘用例:只有一个节点,只有左节点,只有右节点。
(3)无效用例:树为空。
3 思路
分析:
考查二叉树的遍历算法。下面介绍两种解决方式。
(1)层序遍历
对二叉树进行层次遍历(和逐层输出节点问题类似),直接计数即可。
(2)递归实现
利用子节点比根节点深度大1的思想,同时判断左右子树哪个最大取哪个。
4 代码
算法实现:
import java.util.LinkedList;
import java.util.Queue;
public class TreeDepth {
// 1. 层次遍历
public static int maxDepth(TreeNode root) {
if (root == null) {
return 0;
}
int depth = 0;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
depth++;
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
}
return depth;
}
// 2. 利用递归
public static int maxDepth2(TreeNode root) {
if (root == null) {
return 0;
}
int left = maxDepth2(root.left);
int right = maxDepth2(root.right);
return (left > right) ? (left + 1) : (right + 1);
}
public static void main(String[] args) {
// 第一层
TreeNode root = new TreeNode(0);
// 第二层
root.left = new TreeNode(2);
root.right = new TreeNode(4);
// 第三层
root.left.left = new TreeNode(1);
root.right.left = new TreeNode(3);
root.right.right = new TreeNode(-1);
// 第四层
root.left.left.left = new TreeNode(5);
root.left.left.right = new TreeNode(1);
root.right.left.right = new TreeNode(6);
root.right.right.right = new TreeNode(8);
System.out.println(maxDepth(root));
System.out.println(maxDepth2(root));
}
}
参考
在解决本书例题时,参考了一些大佬的题解,比如leetcode上的官方、K神,以及其他的博客,在之后的每个例题详解后都会给出参考的思路或者代码链接,同学们都可以点进去看看!
本例题参考:
本文如有什么不足或不对的地方,欢迎大家批评指正,最后希望能和大家一起交流进步、拿到心仪的 offer !!!