首页 > 代码库 > 求二叉树的最小深度

求二叉树的最小深度

思路:用递归的方法求解。

输入:二叉树的根节点;

输出:二叉树的最小深度。

最小深度的定义:从根节点到叶子节点的最短路径上的节点数。

算法如下

将二叉树分为这么几种情况:

  • 传入的根节点为空,返回NULL;
  • 传入根节点不为空,左子树为空,右子树为空,返回最小深度1;
  • 传入根节点不为空,左子树为空,右子树不为空,返回右子树的最小深度+1;
  • 传入根节点不为空,左子树不为空,右子树为空,返回左子树的最小深度+1;
  • 传入根节点不为空,左右子树都不为空,则返回左右子树中最小深度的较小值+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 NULL;        if(root->left==NULL && root->right == NULL) return 1;        else if(root->left == NULL && root->right !=NULL) return minDepth(root->right)+1;        else if(root->left !=NULL && root->right ==NULL) return minDepth(root->left)+1;        else if(root->left !=NULL && root->right != NULL) return minDepth(root->left)<=minDepth(root->right)?(minDepth(root->left)+1):(minDepth(root->right)+1);    } };