LintCode 二叉树的最小深度

1.描述

给定一个二叉树，找出其最小深度。

二叉树的最小深度为根节点到最近叶子节点的距离。

哪家公司问你的这个题？ Airbnb Alibaba Amazon Apple Baidu Bloomberg Cisco Dropbox Ebay Facebook Google Hulu Intel Linkedin Microsoft NetEase Nvidia Oracle Pinterest Snapchat Tencent Twitter Uber Xiaomi Yahoo Yelp Zenefits

感谢您的反馈

样例

给出一棵如下的二叉树:

        1

     /     \ 

   2       3

          /    \

        4      5  

这个二叉树的最小深度为 2。

2.分析

首先深度为根节点到叶子节点的距离，因此首先需要找到叶子节点，其次每找到一个

根节点到叶子节点的距离判断一下是不是最小的，完成遍历后得到的最小距离即为二叉树

的最小深度。

3.代码

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: The root of binary tree.
     * @return: An integer
     */
    int min=999999;//初始化最小值一个很大的值
    void dfs(TreeNode *root,int height)
    {
        if(root->left==NULL && root->right==NULL)//若为叶子结点
        {
            if(height<min)//若当前距离最小，保留最小的距离替换min
                min=height;
            return ;
        }
        if(root->left!=NULL)//若当前节点还有子树继续向下遍历，深度+1
            dfs(root->left,height+1);
        if(root->right!=NULL)
            dfs(root->right,height+1);
    }
    int minDepth(TreeNode *root) {
        // write your code here
        if(root==NULL)  return 0;
        dfs(root,1);//从根节点开始初始距离为1
        return min;
    }
};

4.总结

若当前节点的左右子树皆为空，则当前节点为叶子节点。只有当找到叶子节点时根节点

到节点的距离才为深度，否则继续向下遍历距离+1。