JZ55 二叉树的深度

描述

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度，根节点的深度视为 1 。

数据范围

• 节点的数量满足：0≤n≤100 ，
• 节点上的值满足0≤val≤100

进阶：

• 空间复杂度 O(1)
• 时间复杂度 O(n)

假如输入的用例为{1,2,3,4,5,#,6,#,#,7}，那么如下图:

示例1

输入：{1,2,3,4,5,#,6,#,#,7}
返回值：4

示例2

输入：{}
返回值：0

解法一：层序遍历(辅助队列queue)

• 定义记录深度deep
• 遍历完一层deep+1
• 通过两层循环
• 第一层while(queue!=null) 进入第一层后获取当前层节点数 size=queue.size()
• 第二层循环 while（size！=null） 出队一个节点，size-1；并让左右节点入队（有的话）

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
import java.util.Queue;
import java.util.LinkedList;
public class Solution {
    public int TreeDepth(TreeNode root) {

        if(root==null){
            return 0;
        }
        
        // 0. 记录深度
        int deep = 0;
        // 1. 辅助队列
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        // 2. 根节入队
        queue.offer(root);
        while (!queue.isEmpty()) {
            // 2.1 记录当前层节点数量
            int size = queue.size();
            // 2.2 当前层左右节点都入队
            while (size != 0) {
                // 根节点出队
                TreeNode node = queue.poll();
                // 2.2 坐边孩子入队(有的话)
                if (node.left != null ) {
                    queue.offer(node.left);
                }
                // 2.2 右边孩子入队（有的话）
                if (node.right != null ) {
                    queue.offer(node.right);
                }
                // 出队一个
                size--;
            }
            // 3. 里层循环走完，深度加1
            deep++;
        }
        return deep;
    }
}

复杂度分析

• 时间复杂度：O(N)
• 空间复杂度：O(N)

解法二：深度遍历(递归)

• 递归在树中最为常见
• 递归求左子树深度
• 递归求右子树深度
• 返回深大的那个加上根节点深度

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
import java.util.Queue;
import java.util.LinkedList;
public class Solution {
    public int TreeDepth(TreeNode root) {

        if (root == null) {
            return 0;
        }
        // 递归求左子树深度
        int leftDeep = TreeDepth(root.left);
        // 递归求右子树深度
        int rightDeep = TreeDepth(root.right);
        // 返回深大的那个加上根节点深度
        return Math.max(leftDeep, rightDeep) + 1;
    }
}

复杂度分析

• 时间复杂度：O(N)
• 空间复杂度：O(N)