剑指Offer对答如流系列 - 二叉树的深度

Posted 2021-03-13 jefferychenxiao

面试题55：二叉树的深度

题目描述

问题（1）二叉树的深度

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

问题（2）平衡二叉树

输入一棵二叉树的根结点，判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

问题分析

问题（1）分析

第一个问题对大家都小意思啦，树的深度=max(左子树深度，右子树深度)+1，采用递归实现。

问题（2）分析

同样需要计算树的深度，树的深度=max(左子树深度，右子树深度)+1。

在遍历过程中，需要判断左右子树深度相差是否超过1，如果不平衡，则令树的深度=-1。最终根据树的深度是否等于-1来确定该树是不是平衡树。

问题解答

问题（1）

     public int TreeDepth(Node root) {
        if(root==null) {
            return 0;
        }
        int left=TreeDepth(root.left);
        int right=TreeDepth(root.right);
        return Math.max(left+1,right+1);
    }

问题（2）

    // 主程序
    public boolean IsBalanced_Solution(Node root) {
        return getDepth(root)!=-1;
    }

    public int getDepth(Node root) {
        if(root==null) {
            return 0;
        }
        int left=getDepth(root.left);
        if(left==-1) {
            return -1;
        }
        int right=getDepth(root.right);
        if(right==-1) {
            return -1;
        }
        return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
    }