代码随想录第十六天-CSDN博客

本文链接：https://blog.csdn.net/zxclong/article/details/135856578

第六章二叉树part03

104.二叉树的最大深度（优先掌握递归）

题目链接/文章讲解/视频讲解： https://programmercarl.com/0104.%E4%BA%8C%E5%8F%89%E6%A0%91%E7%9A%84%E6%9C%80%E5%A4%A7%E6%B7%B1%E5%BA%A6.html

111.二叉树的最小深度（优先掌握递归）

先看视频讲解，和最大深度看似差不多，其实差距还挺大，有坑。

题目链接/文章讲解/视频讲解：https://programmercarl.com/0111.%E4%BA%8C%E5%8F%89%E6%A0%91%E7%9A%84%E6%9C%80%E5%B0%8F%E6%B7%B1%E5%BA%A6.html

思路：

步骤：

定义二叉树节点类TreeNode，包括值val、左子树left和右子树right。
创建BinaryTreeMaxDepth类，其中包含一个maxDepth方法用于计算二叉树的最大深度。该方法采用递归实现，返回左右子树高度的较大值加1。
在maxDepth方法中，先判断当前节点是否为空，若是则表示到达叶子节点，返回0。
递归计算左右子树的高度，并取较大值加1返回，即为当前节点所在子树的最大深度。
在main方法中，构建一颗二叉树，并调用maxDepth方法计算最大深度。
输出结果。
运行程序，得到二叉树的最大深度。

注意事项：

对于一个空树，其最大深度为0。
对于一个只有根节点的树，其最大深度为1。
对于一个非空树，其最大深度为左右子树的最大深度加1。

完整代码

// 定义二叉树节点类
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    public TreeNode(int val) {
        this.val = val;
    }
}

public class BinaryTreeMaxDepth {
    public int maxDepth(TreeNode root) {
        // 递归结束条件，叶子节点的高度为0
        if (root == null) {
            return 0;
        }

        // 递归计算左右子树的高度，并取较大值加1返回
        return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
    }

    public static void main(String[] args) {
        // 构建二叉树
        TreeNode root = new TreeNode(3);
        root.left = new TreeNode(9);
        root.right = new TreeNode(20);
        root.right.left = new TreeNode(15);
        root.right.right = new TreeNode(7);

        BinaryTreeMaxDepth maxDepth = new BinaryTreeMaxDepth();
        System.out.println(maxDepth.maxDepth(root)); // 3
    }
}

222.完全二叉树的节点个数（优先掌握递归）

题目链接/文章讲解/视频讲解：https://programmercarl.com/0222.%E5%AE%8C%E5%85%A8%E4%BA%8C%E5%8F%89%E6%A0%91%E7%9A%84%E8%8A%82%E7%82%B9%E4%B8%AA%E6%95%B0.html

思路：

步骤：

定义完全二叉树节点类TreeNode，包括值val、左子树left和右子树right。
创建CompleteBinaryTreeCount类，其中包含一个countNodes方法用于计算完全二叉树的节点个数。该方法采用递归实现。
在countNodes方法中，先判断当前节点是否为空，若是则返回0。
分别计算左子树和右子树的高度，可以通过两个辅助方法getLeftHeight和getRightHeight实现。
若左右子树的高度相等，则说明是一棵满二叉树，节点个数可以通过公式(1 << leftHeight) - 1计算，其中<<表示位运算左移操作。
若左右子树的高度不相等，则递归计算左右子树的节点个数并相加，再加上根节点1的个数，即为完全二叉树的节点个数。
在main方法中，构建一棵完全二叉树，并调用countNodes方法计算节点个数。
输出结果。
运行程序，得到完全二叉树的节点个数。

注意事项：

对于一个空树，节点个数为0。
对于一棵满二叉树，节点个数可以通过公式(1 << height) - 1计算，其中height为树的高度。
对于一棵非满二叉树，节点个数等于左子树节点个数加右子树节点个数再加上根节点1的个数。

完整代码

// 定义完全二叉树节点类
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    public TreeNode(int val) {
        this.val = val;
    }
}

public class CompleteBinaryTreeCount {
    public int countNodes(TreeNode root) {
        if (root == null) {
            return 0;
        }

        // 分别计算左子树和右子树的高度
        int leftHeight = getLeftHeight(root);
        int rightHeight = getRightHeight(root);

        // 若左右子树的高度相等，则说明是一棵满二叉树，节点个数可以通过公式计算
        if (leftHeight == rightHeight) {
            return (1 << leftHeight) - 1;
        }

        // 若左右子树的高度不相等，则递归计算左右子树的节点个数并相加，再加上根节点1的个数
        return countNodes(root.left) + countNodes(root.right) + 1;
    }

    private int getLeftHeight(TreeNode node) {
        int height = 0;
        while (node != null) {
            height++;
            node = node.left;
        }
        return height;
    }

    private int getRightHeight(TreeNode node) {
        int height = 0;
        while (node != null) {
            height++;
            node = node.right;
        }
        return height;
    }

    public static void main(String[] args) {
        // 构建完全二叉树
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.left = new TreeNode(4);
        root.left.right = new TreeNode(5);
        root.right.left = new TreeNode(6);

        CompleteBinaryTreeCount count = new CompleteBinaryTreeCount();
        System.out.println(count.countNodes(root)); // 6
    }
}