题目描述:
解题1:
这种解法是通过层序遍历去解决这道题。定义一个dep变量代表深度,定义一个队列。开始把根节点加入到队列中,然后通过for循环,把根节点移除,把这层节点的左右子树都加入到队列中,只要队列不为空,就会一直找下一层的子树。每层遍历的时候,深度加1。当队列为空时,证明这颗树全部节点已经遍历完成,此时的深度就是整棵子树的深度,返回深度即可。
class Solution {
public int maxDepth(TreeNode root) {
int dep = 0;
Queue<TreeNode> queue = new LinkedList<>();
if(root != null) queue.add(root);
while(!queue.isEmpty()){
dep++;
for(int i = queue.size(); i > 0; i--){
TreeNode node = queue.poll();
if(node.left != null) queue.offer(node.left);
if(node.right != null) queue.offer(node.right);
}
}
return dep;
}
}
解法2:
这种解法是通过后序遍历去解决。通过递归实现后序遍历,二叉树的深度等于左右子树的最大深度加1。
class Solution {
public int maxDepth(TreeNode root) {
if(root == null) return 0;
return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
}
}