代码随想录day16 Java版

最新推荐文章于 2024-07-22 14:15:06 发布

qq_58958592

最新推荐文章于 2024-07-22 14:15:06 发布

阅读量363

点赞数 10

文章标签： java 深度优先算法

本文链接：https://blog.csdn.net/qq_58958592/article/details/136722545

版权

本文介绍了在LeetCode中解决关于二叉树问题的四种方法：计算最大深度采用递归和层序遍历，最小深度有递归和层序两种版本，以及计算完全二叉树节点个数的简单遍历。

摘要由CSDN通过智能技术生成

最近要搞论文了，课程也比较忙，拖了几天。

力扣104 二叉树最大深度

思路简单，层序或者递归做都行

// 递归
class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null) return 0;
        return dfs(root);
    }
    public int dfs(TreeNode root) {
        if (root == null) return 0;
        return Math.max(dfs(root.left), dfs(root.right)) + 1;
    }
}
// 层序
class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null) return 0;
        int depth = 0;
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);
        while (!q.isEmpty()) {
            int size = q.size();
            while (size -- > 0) {
                TreeNode temp = q.poll();
                if (temp.left != null) q.add(temp.left);
                if (temp.right != null) q.add(temp.right);
            }
            depth ++;
        }
        return depth;
    }
}

力扣111 二叉树的最小深度

// 层序
class Solution {
    public int minDepth(TreeNode root) {
        int depth = 1;
        if (root == null) return 0;
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);
        while (!q.isEmpty()) {
            int size = q.size();
            while (size -- > 0) {
                TreeNode temp = q.poll();
                if (temp.left == null && temp.right == null) return depth;
                if (temp.left != null) q.add(temp.left);
                if (temp.right != null) q.add(temp.right);
            }
            depth ++;
        }
        return depth;
    }
}
// 递归
class Solution {
    public int minDepth(TreeNode root) {
        if (root == null) return 0;
        return dfs(root);
    }
    public int dfs(TreeNode root) {
        if (root == null) return 100000;
        else if (root.left == null && root.right == null) return 1;
        return Math.min(dfs(root.left), dfs(root.right)) + 1;
    }
}
// 正常递归
class Solution {
    public int minDepth(TreeNode root) {
        if (root == null) return 0;
        int leftHeight = minDepth(root.left);
        int rightHeight = minDepth(root.right);
        if (root.left != null && root.right == null) {
            return 1 + leftHeight;
        }
        if (root.right != null && root.left == null) {
            return 1 + rightHeight;
        }
        return 1 + Math.min(leftHeight, rightHeight);
    }
}

力扣222 完全二叉树的节点个数

每遍历到一个节点就加一即可，选择什么遍历方式都可。

class Solution {
    public int countNodes(TreeNode root) {
        if (root == null) return 0;
        int leftCount = countNodes(root.left);
        int rightCount = countNodes(root.right);
        return leftCount + rightCount + 1;
    }
}