《剑指offer》之二叉树的深度

程序员爱酸奶

发布于 2020-03-13 10:19:22

3880

发布于 2020-03-13 10:19:22

文章被收录于专栏：程序员爱酸奶程序员爱酸奶程序员爱酸奶

前言

我们今天接着来看一道关于二叉树的算法题，关于二叉树的深度。

题目

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

分析

求该树的深度，主要就是看最长路径。比如下图的深度为5，最长的路径为34，99，35，64，77

那应该怎么做？这里用递归，如果当前节点没有左右节点，就返回当前节点，如果有左右节点，就返回左右节点的，比较左节点和右节点的深度，谁的深度大就返回那个。这样就可以获得树的最大深度啦。

解法

public int TreeDepth(TreeNode root) {
        if(root==null){
            return 0;
        }

        int left=TreeDepth(root.left);
        int right=TreeDepth(root.right);
        if(left>right){
            return left+1;
        }
        return right+1;

    }

上面主要注意的是left+1 和right+1;为什么要加一呢，因为我们递归的出口是当前节点为null ,返回0，为1个节点的话返回1.

测试

测试main方法

public static void main(String[] args) {
        TreeNode root =new TreeNode(34);
        root.left=new TreeNode(23);
        root.right=new TreeNode(99);
        root.left.left=new TreeNode(1);
        root.left.right=new TreeNode(27);
        root.right.left=new TreeNode(35);
        root.right.left.right=new TreeNode(64);
        root.right.left.right.right=new TreeNode(77);
        TreeOperation.show(root);
        Solution solution= new Solution();
        System.out.println(solution.TreeDepth(root));
    }