【leetcode】111. 二叉树的最小深度（minimum-depth-of-binary-tree）（BFS）[简单]

链接

https://leetcode-cn.com/problems/minimum-depth-of-binary-tree/

耗时

解题：12 min
题解：3 min

题意

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

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

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

思路

BFS 层序遍历，在遇到第一个叶节点时，返回当前层数。

时间复杂度：$O(n)$ n 为二叉树节点数量

AC代码

/**
 * 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 == NULL) return 0;
        
        queue<TreeNode*> q;
        q.push(root);
        int ans = 0;
        while(!q.empty()) {
            ans++;
            int n = q.size();
            for(int i = 0; i < n; ++i) {
                TreeNode* now = q.front();
                q.pop();
                if(now->left == NULL && now->right == NULL) {
                    return ans;
                }
                if(now->left != NULL) q.push(now->left);
                if(now->right != NULL) q.push(now->right);
            }
        }
        
        return ans;
    }
};