leetcode 111. 二叉树的最小深度

问题描述

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

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

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

示例:

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

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

代码

这道题和二叉树的层次遍历使用的一套模板：

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

结果：

执行用时 :16 ms, 在所有 C++ 提交中击败了62.83%的用户
内存消耗 :22.2 MB, 在所有 C++ 提交中击败了5.16%的用户