111. 二叉树的最小深度

题目描述

给定一个二叉树，找出其最小深度。最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

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

 

示例

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

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

 

思路

dfs分为3种情况：

（1）若树为空，则深度为0；

（2）若树只有一个根结点，则深度为1；

（3）若树只有左子树或者右子树，则返回左子树或右子树的高度

（4）否则，返回左右子树较小的高度，最后加1。

 

代码

/**
 * 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;
        if (!root->left && !root->right) return 1;
        if (!root->left) return 1 + minDepth(root->right);
        if (!root->right) return 1 + minDepth(root->left);
        return 1 + min(minDepth(root->left), minDepth(root->right));
    }
};

 

思路2

层序遍历。记录当前层数。

 

代码

class Solution {
public:    
    int minDepth(TreeNode* root) {
        if(root == nullptr) return 0;  
        queue<pair<TreeNode*,int>> q;  //把节点和所在的层数绑定
        q.push(make_pair(root,1)); //压入根节点，层数为1
        while(1){
            TreeNode* node = q.front().first; //当前节点
            int level = q.front().second;  //层数
            q.pop();
            if (!node->left && !node->right) return level;//遇到叶子节点
            if(node->left) //压入左节点
                q.push(make_pair(node->left, level+1));
            if(node->right)//压入右节点
                q.push(make_pair(node->right, level+1));
        }
    }
};