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

  • 2020 年 3 月 13 日
  • 笔记

前言

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

题目

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

分析

求该树的深度,主要就是看最长路径。比如下图的深度为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));      }