目录

  • 前言
  • 题目描述
  • 讲解
  • 一、递归
  • 二、非递归


前言

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

题目描述

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

讲解

一、递归

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

java递归太深了导致异常 java递归深度_二叉树

递归代码比较简单,但是我们还是来分析分析调用过程,着重讲左子树调用过程,右子树调用过程相同:

java递归太深了导致异常 java递归深度_算法_02

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

java递归太深了导致异常 java递归深度_队列_03