欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

【剑指offer简单部分】二叉树的深度(java通俗解释)

程序员文章站 2022-07-10 17:26:47
目录前言题目描述一、递归二、非递归前言二叉树的深度有递归和非递归两种做法。题目描述输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。一、递归代码看起来简单,主要理解递归调用过程,有句玩笑叫" 递归过程靠想象 "递归思路:1、递归跳出条件是判断一个节点是否是空,如果为空则跳出2、如果不为空则加一继续递归3、最后的返回值只需返回左子树和右子树中的最大值。public class Solution { public...

前言

二叉树的深度有递归和非递归两种做法。

题目描述

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

讲解

一、递归

代码看起来简单,主要理解递归调用过程,有句玩笑叫" 递归过程靠想象 ",说明递归比较抽象
递归思路:
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);
    }
}

【剑指offer简单部分】二叉树的深度(java通俗解释)

递归代码比较简单,但是我们还是来分析分析调用过程,着重讲左子树调用过程,右子树调用过程相同:【剑指offer简单部分】二叉树的深度(java通俗解释)

总结起来就是,先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;
    }
}

【剑指offer简单部分】二叉树的深度(java通俗解释)

本文地址:https://blog.csdn.net/weixin_39615182/article/details/110221216