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;
    }
}

难点

• 递归法中单层递归的逻辑

总结

巩固了递归法和迭代法。