JZ55 二叉树的深度

描述

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

示例1
输入：
{1,2,3,4,5,#,6,#,#,7}
返回值：
4

解题思路

1.递归

遇见二叉树第一时间想到递归，话不多说，代码如下：

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public int TreeDepth(TreeNode root) {
        //1.递归
        if(root == null) return 0;
        return Math.max(TreeDepth(root.left), TreeDepth(root.right)) + 1;
    }
}

2.迭代

层序遍历一遍，用一个变量level记录层数即可，可以采用迭代的思路，利用队列实现。

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.Queue;
import java.util.LinkedList;
public class Solution {
    public int TreeDepth(TreeNode root) {
        //2.迭代，层序遍历记录层数即可
        Queue<TreeNode> queue = new LinkedList<>();
        if(root != null) queue.offer(root);
        int level = 0;
        while(!queue.isEmpty()){
            int capacity = queue.size();
            while(capacity -- > 0){
                TreeNode node = queue.poll();
                if(node.left != null) queue.offer(node.left);
                if(node.right != null) queue.offer(node.right);
            }
            level ++;
        }
        return level;
    }
}