LeetCode 111 二叉树的最小深度

题目

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

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

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

示例:

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

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

算法思路1

使用递归来算出二叉树的最小深度，深度优先DFS考虑
这道题的关键是搞清楚递归结束条件

• 叶子节点的定义是左孩子和右孩子都为 null 时叫做叶子节点
• 当 root 节点左右孩子都为空时，返回 1
• 当 root 节点左右孩子有一个为空时，返回不为空的孩子节点的深度
• 当 root 节点左右孩子都不为空时，返回左右孩子较小深度的节点值

代码

class Solution {
    public int minDepth(TreeNode root) {
        if (root == null)   //递归终止条件
            return 0;
        if(root.left == null && root.right == null)  //找到叶子节点
            return 1;
        int m1 = minDepth(root.left);   //
        int m2 = minDepth(root.right);    
        if(root.right == null || root.left == null)   //左右孩子一个为空，那么m1 和m2 有一个必为0，返回不空孩子那个的深度                
            return m1+m2+1;
            
         
        return Math.min(m1,m2)+1;   //都不空，返回深度较小的孩子的深度
        
        
    }
}

复杂度分析

时间复杂度：我们访问每个节点一次，时间复杂度为 O(N) ，其中 N 是节点个数。
空间复杂度：最坏情况下，整棵树是非平衡的，例如每个节点都只有一个孩子，递归会调用 N （树的高度）次，因此栈的空间开销是 O(N)。但在最好情况下，树是完全平衡的，高度只有 log(N)，因此在这种情况下空间复杂度只有 O(log(N))。

算法思路2

使用广度优先，BFS来考虑
我们按照树的层去迭代，第一个访问到的叶子就是最小深度的节点，这样就不用遍历所有的节点了。

• 使用队列来记录二叉树的元素
• 用一个标记来标记同一层，对同一深度（层）的节点一起处理，判断其中是否有叶子节点，有则返回当前层的深度。

代码

class Solution {
    public int minDepth(TreeNode root) {
        if (root == null) return 0;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int depth = 1;  //根结点有，一开始深度为1
        while (!queue.isEmpty()) {
            int qSize = queue.size();  //标记为同一层的元素
            for (int i=0; i < qSize; i++) {  //对同一层的元素判断
                TreeNode delNode = queue.remove();
                if (delNode.left == null && delNode.right == null) return depth;  //到叶子节点，直接返回当前深度
                if (delNode.left != null) queue.add(delNode.left);
                if (delNode.right != null) queue.add(delNode.right);
            }
            depth++;  //这一层都有孩子结点，就层数+1，继续判断队列中拿出的元素（是这一层结点的孩子结点）
        }
        return -1; //正常不会走到这
    }
}

复杂度分析

时间复杂度：最坏情况下，这是一棵平衡树，我们需要按照树的层次一层一层的访问完所有节点，除去最后一层的节点。这样访问了 N/2个节点，因此复杂度是 O(N)。
空间复杂度：和时间复杂度相同，也是 O(N)。