111. Minimum Depth of Binary Tree

111. 二叉树的最小深度

题目描述

求二叉树最小深度，也就是从根节点到最近叶子节点的最短路径上的节点数量。

思路

层序遍历二叉树，记录最小深度。

改进

• 最快的做法
  使用了一个map记录节点和自己的深度。

class Solution {
public:
    int minDepth(TreeNode *root) {
        if (root == nullptr) {
            return 0;
        }
        queue<pair<TreeNode *, int> > que;//<节点,深度>
        que.emplace(root, 1);
        while (!que.empty()) {
            TreeNode *node = que.front().first;
            int depth = que.front().second;
            que.pop();
            if (node->left == nullptr && node->right == nullptr) {
                return depth;
            }
            if (node->left != nullptr) {
                que.emplace(node->left, depth + 1);
            }
            if (node->right != nullptr) {
                que.emplace(node->right, depth + 1);
            }
        }

        return 0;
    }
};

• 最省空间的做法

class Solution {
public:
    int minDepth(TreeNode* root) {
        int res = 0;
        if (!root){
            return res;
        }
        queue<TreeNode*> q;//节点队列
        q.push(root);
        ++res;
        while (!q.empty()){
            int n = q.size();
            while (n--){
                TreeNode* r = q.front();
                q.pop();
                if (!r->left && !r->right){
                    return res;
                }
                if (r->left){
                    q.push(r->left);
                }
                if (r->right){
                    q.push(r->right);
                }
            }
            ++res;
        }
        return res;
    }
};