【剑指offer简单部分5】二叉树的深度（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;
    }
}