剑指offer【08】- 二叉树的深度(java)

题目：二叉树的深度

考点：知识迁移能力

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

牛客网上的剑指offer题，我按从简单到难的排序来做了

思路：从根节点出发， 查询左子树的深度 ， 获取右子树的深度，比较一下，取大的，再加一 。就是整个二叉树的深度

 

法一：递归

/**
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) {
        //递归的出口，如果节点没有值，则返回0
        if(root == null){
            return 0;
        }
        //获取左子树的最大深度
        int leftDep = TreeDepth(root.left);
        //获取右子树的最大深度
        int rigthDep = TreeDepth(root.right);
        //找出最大值，并且加上root这一层即可
        return Math.max(leftDep,rigthDep)+1;
    }
}

法二：非递归法，层次遍历法

/**
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) {
        if(root == null){
            return 0;
        }
        int count = 0, depth = 0, nextCount = 1;

        Queue<TreeNode> queue = new LinkedList<TreeNode>();

        //将根节点添加到队列中
        queue.add(root);

        //第一次循环时队列的长度为1
        while(queue.size()!=0){
            count++;
            //先进先出，取出队列的第一个元素
            TreeNode top = queue.poll();
            //如果根节点的左子树不为空，则将左子树的根节点添加到队列中
            if(top.left!=null){
                queue.add(top.left);
            }
             //如果根节点的右子树不为空，则将右子树的根节点添加到队列中
             if(top.right!=null){
                 queue.add(top.right);
            }
            //当同一层的节点全部添加到队列中时，count与nextCount相等，deph+1
            if(count==nextCount){
                nextCount = queue.size();
                depth++;
                count=0;
            }
        }
        return depth;

    }
}