题目描述
求给定二叉树的最小深度。最小深度是指树的根结点到最近叶子结点的最短路径上结点的数量。
Given a binary tree, find its minimum depth.The minimum depth is the number of nodes along the shortest path from the root node down to the nearest leaf node.
示例1
输入
{1, 2, 3, 4, 5}
输出
{2}
解题思路
陷阱
如果根结点的左子树为空、右子树不为空(或者右子树为空、左子树不为空)时,则返回右子树的最小深度 + 1(根结点)(或者左子树的最小深度 + 1)。
思路
- 如果根结点为空,则返回0
- 如果根结点的左子树为空、右子树也为空,则返回1
- 如果根结点的左子树为空、右子树不为空,则返回右子树的最小深度并加1
- 如果根节点的右子树为空、左子树不为空,则返回左子树的最小深度并加1
- 如果根节点的左子树不为空、右子树也不为空,则返回左、右子树最小深度中更小的那个深度并加1
代码实现
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
class Solution {
public:
/**
* 求给定二叉树的最小深度
* @param root TreeNode类 二叉树的根结点
* @return int整型 二叉树的最小深度
*/
int run(TreeNode* root) {
// write code here
if(root == NULL) return 0;
if(root->left == NULL && root->right == NULL) return 1;
if(root->left == NULL && root->right != NULL) return run(root->right) + 1;
if(root->left != NULL && root->right == NULL) return run(root->left) + 1;
int leftSubtreeMinimumDepth = run(root->left);
int rightSubtreeMinimumDepth = run(root->right);
return leftSubtreeMinimumDepth < rightSubtreeMinimumDepth
? leftSubtreeMinimumDepth + 1 :rightSubtreeMinimumDepth + 1;
}
};
运行结果
运行时间:4ms
占用内存:780k