【剑指offer简单部分】二叉树的深度(java通俗解释)
前言
二叉树的深度有递归和非递归两种做法。
题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
讲解
一、递归
代码看起来简单,主要理解递归调用过程,有句玩笑叫" 递归过程靠想象 ",说明递归比较抽象
递归思路:
1、递归跳出条件是判断一个节点是否是空,如果为空则跳出
2、如果不为空则加一继续递归
3、最后的返回值只需返回左子树和右子树中的最大值。
public class Solution {
public int TreeDepth(TreeNode root) {
if(root == null){
return 0;
}
int left = TreeDepth(root.left)+1;
int right = TreeDepth(root.right)+1;
return Math.max(left,right);
}
}
递归代码比较简单,但是我们还是来分析分析调用过程,着重讲左子树调用过程,右子树调用过程相同:
总结起来就是,先root指向A节点,然后先一直往左边下面进行调用,再往回递归,回到A节点时,左边调用和递归完毕,即A节点(第一层)int left=TreeDepth(root.left)+1,执行完毕,得到A的left值,再继续int right = TreeDepth(root.right)+1,也就是A节点右边的调用和递归,最后递归回到A,就可以得到A节点的right值,最后比较A节点左边最深的层(left记录左边最深),和右边最深的层(right记录右边最深),取最大值,也就是整个二叉树最深的层,即二叉树的深度。
二、非递归
非递归思路:
非递归采用二叉树的层序遍历的思想,层序遍历用队列。我们需要遍历每层,每遍历完一层则将MaxDepth++。先将根节点入队,然后判断队列是否为空,非空则先获取队列的大小size,也就是队列中节点个数,再用循环将队列中的所有节点出队,之后将所有节点的左右孩子入队,这样就可以达到将每一层的节点入队的效果。非递归代码和思路还是挺一直,比较直观,也较容易理解。
代码如下:
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
public int TreeDepth(TreeNode root) {
if(root == null){
return 0;
}
int MaxDepth = 0;
TreeNode cur = null; //用于指向出队元素
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()){
MaxDepth++;
int size = queue.size();
for (int i = 0; i <size; i++) {
cur = queue.peek(); //返回出队元素
queue.poll(); //出队(删除)
if (cur.left != null)
queue.offer(cur.left);
if (cur.right != null)
queue.offer(cur.right);
}
}
return MaxDepth;
}
}
本文地址:https://blog.csdn.net/weixin_39615182/article/details/110221216
上一篇: Java入门 Java概述
下一篇: Java Collection使用介绍