LeetCode 111. 二叉树的最小深度

1. 题目信息

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

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明: 叶子节点是指没有子节点的节点。

示例:

给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回它的最小深度  2.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-depth-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

2.1 DFS

class Solution {
public:
    int minDepth(TreeNode* root) {
        if(root == NULL)
            return 0;
        int leftHeight, rightHeight;
        if(root->left && root->right)
        {
            leftHeight = minDepth(root->left);
            rightHeight = minDepth(root->right);
        }
        else if(root->left && !root->right)
        {
            leftHeight = minDepth(root->left);
            rightHeight = INT_MAX;
        }
        else if(!root->left && root->right)
        {
            leftHeight = INT_MAX;
            rightHeight = minDepth(root->right);
        }
        else
        {
            leftHeight = rightHeight = 0;
        }
        return min(leftHeight, rightHeight)+1;
    }
};

2.2 BFS

按层遍历，遇见第一个叶子节点（左右子节点均为NULL）时，退出

class Solution {//BFS
public:
    int minDepth(TreeNode* root) {
        if(root == NULL)
            return 0;
        int height = 0, n;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
            height++;
            n = q.size();
            while(n--)
            {
                if(!q.front()->left && !q.front()->right)//第一个叶子节点退出
                    return height;
                if(q.front()->left)
                    q.push(q.front()->left);
                if(q.front()->right)
                    q.push(q.front()->right);
                q.pop();
            }
        }
        return height;
    }
};