代码随想录-二叉树 | 111 二叉树的最小深度

LeetCode 111 二叉树的最小深度
- 解题思路
- 代码
- 难点
- 总结

LeetCode 111 二叉树的最小深度

题目链接
代码随想录

题目描述

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

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

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

解题思路

判断：

递归法
- 确定递归函数的参数和返回值：参数-根节点，返回值-最小深度；
- 确定终止条件：节点为空，返回0，表示当前高度为0；
- 确定单层递归的逻辑：判断是否为叶子节点，若不是，
  - 左子树为空：最小深度 = 1 + 右子树最小深度；
  - 右子树为空：最小深度 = 1 + 左子树最小深度。
迭代法：层序遍历
- 终止条件：当左右孩子都为空时，说明遍历到了最低点。

代码

递归法

class Solution {
    public int minDepth(TreeNode root) {
        if(root == null) return 0;

        int leftDepth = minDepth(root.left);
        int rightDepth = minDepth(root.right);
        if(root.left == null) return rightDepth + 1;
        if(root.right == null) return leftDepth + 1;

        //左右节点都不为null
        return Math.min(leftDepth, rightDepth) + 1;
    }
}

迭代法

class Solution {
    public int minDepth(TreeNode root) {
        if(root == null) return 0;

        Deque<TreeNode> deque = new LinkedList<>();
        deque.offer(root);
        int depth = 0;
        while(!deque.isEmpty()){
            depth++;
            int size = deque.size();
            for(int i = 0; i < size; i++){
                TreeNode node = deque.poll();
                if(node.left == null && node.right == null) return depth;
                if(node.left != null) deque.offer(node.left);
                if(node.right != null) deque.offer(node.right);
            }
        }
        return depth;
    }
}