欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【LeetCode】104. Maximum Depth of Binary Tree 二叉树的深度 DFS BFS 递归方式 迭代方式 JAVA

程序员文章站 2024-01-11 18:55:10
...

前言

这次的题目是二叉树的深度遍历,总体上来说吧,难度没有那么大,可是我就是再迭代的地方爬不出来了,有些题解也没有注释,讲解的也不是很清楚,所以就看起来有点麻烦

题目传送门: 点击此处

题目截图:
【LeetCode】104. Maximum Depth of Binary Tree 二叉树的深度 DFS BFS 递归方式 迭代方式 JAVA

递归方式
public int maxDepth(TreeNode root) {
    return root == null ? 0 : Math.max(maxDepth(root.left) + 1, maxDepth(root.right) + 1);
}

PS:
虽然有人不是很理解代码为什么要写成一行,但是我觉得写成一行很得劲儿
虽然会被嫌弃,但是我还是想这么干
虽然我也嫌弃别人这么些,但是我也还是像这样写

迭代方式

BFS

不要吐槽我,注释稍后添加!!!!!!!!!!!!!!!!!!!!

public int maxDepth(TreeNode root) {
    if(root == null) {
        return 0;
    }
    Stack<TreeNode> stack = new Stack<>();
    Stack<Integer> value = new Stack<>();
    stack.push(root);
    value.push(1);
    int max = 0;
    while(!stack.isEmpty()) {
        TreeNode node = stack.pop();
        int temp = value.pop();
        max = Math.max(temp, max);
        if(node.left != null) {
            stack.push(node.left);
            value.push(temp+1);
        }
        if(node.right != null) {
            stack.push(node.right);
            value.push(temp+1);
        }
    }
    return max;
}

DFS

public int maxDepth(TreeNode root) {
    if(root == null) {
        return 0;
    }
    Queue<TreeNode> queue = new LinkedList<>();
    queue.offer(root);
    int count = 0;
    while(!queue.isEmpty()) {
        int size = queue.size();
        while(size > 0) {
            TreeNode node = queue.poll();
            size--;
            if(node.left != null) {
                queue.offer(node.left);
            }
            if(node.right != null) {
                queue.offer(node.right);
            }
        }
        count++;
    }
    return count;
}