二叉树最大深度
如果二叉树为空,则深度为0
如果不为空,分别求左子树的深度和右子树的深度,去最大的再加1,因为根节点深度是1,要加进去。
class Solution {
public:
/**
* @param root: The root of binary tree.
* @return: An integer
*/
int maxDepth(TreeNode * root) {
if(!root)
return 0;
int leftHeight=maxDepth(root->left);
int rightHeight=maxDepth(root->right);
return max(leftHeight,rightHeight)+1;
}
};
二叉树最小深度
刚开始拿到这道题,就想不是很简单吗,不就是最大深度的代码改为最小就行了吗? 这样做的结果就是有些测试时通不过的。
因为如果出现斜树的情况,也就是说只有左子树,或只有右子树的时候,如果按照刚才说的那个算法来实现的话,得出的答案是0,明显是错误的。
一种实现方法,一种就是计算左子树和右子树深度的时候,判断是否等于0,如果等于0,说明该子树不存在,深度赋值为最大值。
class Solution {
public:
/**
* @param root: The root of binary tree
* @return: An integer
*/
int minDepth(TreeNode * root) {
// write your code here
if(!root)
return 0;
if(!root->left && !root->right)
return 1;
int leftHeight=minDepth(root->left);
if(leftHeight==0)
leftHeight=INT32_MAX;
int rightHeight=minDepth(root->right);
if(rightHeight==0)
rightHeight=INT32_MAX;
return min(leftHeight,rightHeight)+1;
}
};