题目:二叉树的最大深度
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
package leetCode.DFS;
import java.util.LinkedList;
public class lc_dfs_104_maxDepth {
/*
思路:本题和剑指offer38题一样,这回我两个方法都做出来了
复制思路如下:
1)递归
2)层次遍历
*/
//递归方式:总深度=左右子树中较大的深度+1
public int maxDepth(TreeNode root) {
if (root == null)
return 0;//叶子结点的下一层为null,null层计为0深度
int leftDep = maxDepth(root.left);
int rightDep = maxDepth(root.right);
return Math.max(leftDep, rightDep) + 1;//+1表示加上当前层的深度
}
//非递归方式,使用队列层次遍历树,计算每一层的结点数,每层计算完深度+1,
//其实就是从上到下从左到右遍历二叉树的改进
public int maxDepth2(TreeNode root) {
if (root == null)
return 0;
LinkedList<TreeNode> queue = new LinkedList<>();
//将根节点入栈
queue.offer(root);
int depth = 0;
int curCount = 0;//记录遍历到当前层的第几个结点
int count = 1;//当前层的结点数,初始为第一层,只有根节点,所以是1
while (!queue.isEmpty()) {
//取出当前队列的第一个结点top
TreeNode top = queue.poll();
curCount++;
if (top.left != null)
queue.offer(top.left);
if (top.right != null)
queue.offer(top.right);
//若没有达到count就不断从队列头取出,直到当前层结点遍历完
if (curCount == count) {//当前层结点遍历完
depth++;//深度+1,即将进入下层
count = queue.size();//计算下一层要遍历的结点数,即目前队列内的结点数
curCount = 0;//重置下一轮的计数器
}
}
return depth;
}
}