二叉树的最大深度-leetcode刷题记录四

235 阅读1分钟

给定一个二叉树,找出其最大深度。

二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

说明: 叶子节点是指没有子节点的节点。

示例: 给定二叉树 [3,9,20,null,null,15,7]

题目链接

广度优先算法

    public int maxDepth(TreeNode root) {
        //空树,返回0
        if (root == null) {
            return 0;
        }
        //记录每一层节点的队列
        Deque<TreeNode> treeNodes = new LinkedList<>();
        //根节点入队
        treeNodes.offer(root);
        int count = 0;
        while (!treeNodes.isEmpty()) {
            //这里的广度优先算法是一层遍历完再去遍历下一层,所以需要一个size来记录当前队列里面需要遍历的节点数量
            int size = treeNodes.size();
            while (size>0){
                TreeNode treeNode = treeNodes.poll();
                if (treeNode.left!=null){
                    treeNodes.offer(treeNode.left);
                }
                if (treeNode.right!=null){
                    treeNodes.offer(treeNode.right);
                }
                size--;
            }
            //遍历完一层之后计数+1
            count++;
        }
        return count;
    }

递归

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