104. 二叉树的最大深度
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
解法
遍历的时候带上当前节点的深度即可。
class Solution {
public int maxDepth(TreeNode root) {
if (root == null){
return 0;
}
return traversal(root, 0);
}
public int traversal(TreeNode node, int depth) {
if (node == null){
return depth;
}
depth++;
int leftDepth = traversal(node.left, depth);
int rightDepth = traversal(node.right, depth);
return Math.max(leftDepth, rightDepth);
}
}
优化
class Solution {
public int maxDepth(TreeNode root) {
if (root == null){
return 0;
}
int leftDepth = maxDepth(root.left);
int rightDepth = maxDepth(root.right);
//加1的意思是当前节点加上左右子树的最大深度为整体最大深度
return Math.max(leftDepth, rightDepth) + 1;
}
}