题目描述:
求二叉树的最小深度,即从根节点到叶节点的最短距离。
思路:
使用递归,不断判断当前节点左子树和右子树是否为空,都为空则到达叶节点,当前距离加1。左子树右子树不为空则求左子树或者右子树的深度,求最小的。
代码如下:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int minDepth(TreeNode *root) {
if(root==NULL){
return 0;
}
return minDepTree(root,0);
}
int minDepTree(TreeNode *root,int sum){
if(root->left==NULL&&root->right==NULL){
return sum+1;
}
else if(root->left==NULL&&root->right!=NULL){
return minDepTree(root->right,sum)+1;
}
else if(root->left!=NULL&&root->right==NULL){
return minDepTree(root->left,sum)+1;
}
else{
return min(minDepTree(root->left,sum)+1,minDepTree(root->right,sum)+1);
}
}
};