给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明:叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回它的最大深度 3。
分析
对于根节点来说,如果左子树的深度大于右子树的深度,则整棵树的深度是左子树深度 + 1(根节点),反之整棵树的深度是右子树深度 + 1(根节点)。
那么每次递归要做的事情就是分别计算左子树的深度和右子树的深度,并且做比较取最大值,最后树的深度就是最大值 + 1
完整递归算法
public int maxDepth(TreeNode root) {
if(root == null) {
return 0;
} else {
int leftDepth = maxDepth(root.left);
int rightDepth = maxDepth(root.right);
return Math.max(leftDepth, rightDepth) + 1;
}
}
递归转迭代
递归转迭代可以按照下方模式处理:
private int maxDepth(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
int depth = 0;
while(!queue.isEmpty()) {
// TODO 每次循环需要做的事情
depth++;
}
return depth;
}
现在关键是分析出每次循环需要做的事情!
拿示例中的二叉树来分析
3
/ \
9 20
/ \
15 7
队列:3
取出队列中的头元素 3
放入 9,20
完整迭代算法
public int maxDepth(TreeNode root) {
if (root == null) {
return 0;
}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
int ans = 0;
while (!queue.isEmpty()) {
int size = queue.size();
while (size > 0) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
size--;
}
ans++;
}
return ans;
}