2021-04-07

104.二叉树的最大深度

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

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

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

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

    3
   / \
  9  20
    /  \
   15   7
返回它的最大深度 3 。

题解1:深度优先搜索(DFS)

import javax.swing.tree.TreeNode;

/*
 * @lc app=leetcode.cn id=104 lang=java
 *
 * [104] 二叉树的最大深度
 */

// @lc code=start
/**
 * Definition for a binary tree node. public class TreeNode { int val; TreeNode
 * left; TreeNode right; TreeNode() {} TreeNode(int val) { this.val = val; }
 * TreeNode(int val, TreeNode left, TreeNode right) { this.val = val; this.left
 * = left; this.right = right; } }
 */
class Solution {
    public int maxDepth(final TreeNode root) {
        // DFS:深度优先算法
        // 时间复杂度:O(n)
        // 空间复杂度:O(height)
        if (root == null) {
            return 0;
        } else {
            int left = maxDepth(root.left);
            int right = maxDepth(root.right);
            return Math.max(left, right) + 1;
        }
    }
}

题解2:广度优先搜索（BFS）

import java.util.LinkedList;
import java.util.Queue;

import javax.swing.tree.TreeNode;

/*
 * @lc app=leetcode.cn id=104 lang=java
 *
 * [104] 二叉树的最大深度
 */

// @lc code=start
/**
 * Definition for a binary tree node. public class TreeNode { int val; TreeNode
 * left; TreeNode right; TreeNode() {} TreeNode(int val) { this.val = val; }
 * TreeNode(int val, TreeNode left, TreeNode right) { this.val = val; this.left
 * = left; this.right = right; } }
 */
class Solution {
    public int maxDepth(final TreeNode root) {
        // BFS:广度优先算法
        // 时间复杂度:O(n)
        // 空间复杂度:O(n)
        if (root == null) {
            return 0;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        int ans = 0;
        while (!queue.isEmpty()) {
            int size = queue.size();
            // 遍历当前层所有节点，并且找到下一层所有节点加入到队列
            while (size > 0) {
                TreeNode curNode = queue.poll();
                if (curNode.left != null) {
                    // 左叶子不为null,加入队列
                    queue.offer(curNode.left);
                }
                if (curNode.right != null) {
                    // 右叶子不为null,加入队列
                    queue.offer(curNode.right);
                }
                size--;
            }
            // 总深度+1
            ans++;
        }
        return ans;
    }
}