Leetcode 111. 二叉树的最小深度

Leetcode 111. 二叉树的最小深度

原题链接

广度优先

class Solution {
    // 内部类，记录节点信息和当前节点的深度
    static class QueueNode {
        TreeNode treeNode;
        int depth;

        public QueueNode(TreeNode node, int depth) {
            this.treeNode = node;
            this.depth = depth;
        }
    }

    public int minDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        // 使用队列记录节点信息
        Queue<QueueNode> queue = new LinkedList<>();
        queue.offer(new QueueNode(root, 1));
        while (!queue.isEmpty()) {
            QueueNode queueNode = queue.poll();
            TreeNode treeNode = queueNode.treeNode;
            int depth = queueNode.depth;
            // 如果treeNode的左节点与右节点都未空，则当前节点的深度就是整棵树的最小深度
            if (treeNode.left == null && treeNode.right == null) {
                return depth;
            }

            if (treeNode.left != null) {
                queue.offer(new QueueNode(treeNode.left, depth + 1));
            }
            if (treeNode.right != null) {
                queue.offer(new QueueNode(treeNode.right, depth + 1));
            }
        }
        return 0;
    }
}

深度优先

class Solution {
    public int minDepth(TreeNode root) {
        return dfs(root);
    }

    private int dfs(TreeNode root) {
        if (root == null) {
            return 0;
        }
        // 没有子节点，返回当前层
        if (root.left == null && root.right == null) {
            return 1;
        }
        int minDepth = Integer.MAX_VALUE;
        if (root.left != null) {
            // 递归计算左子树的最小深度
            minDepth = Math.min(dfs(root.left), minDepth);
        }
        if (root.right != null) {
            // 递归计算右子树的最小深度
            minDepth = Math.min(dfs(root.right), minDepth);
        }
        // 返回最小深度（返回时需要加上当前层）
        return minDepth + 1;
    }
}