题目描述:
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回它的最小深度 2.
思路:
递归
判断每种情况的返回值
代码如下:
/**
* Definition for a binary tree node.
* 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) return 0;
if(root->left!=NULL&&root->right!=NULL) return min(minDepth(root->left),minDepth(root->right))+1;
if(root->left==NULL&&root->right==NULL) return 1;
if(root->left!=NULL&&root->right==NULL) return minDepth(root->left)+1;
if(root->left==NULL&&root->right!=NULL) return minDepth(root->right)+1;
return 0;
}
};
Python:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def minDepth(self, root: TreeNode) -> int:
if not root: return 0
if root.left and root.right: return min(self.minDepth(root.left),self.minDepth(root.right))+1
if not root.left and not root.right: return 1
if not root.left and root.right: return self.minDepth(root.right)+1
if root.left and not root.right: return self.minDepth(root.left)+1
return 0