二叉树的深度

题目描述

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

源代码：

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

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

    }

}
*/
public class Solution {
    public int max(int a,int b){
        return a > b ? a:b;
    }
    public int TreeDepth(TreeNode root) {
        if(root==null) return 0;
        return max(1+TreeDepth(root.left),1+TreeDepth(root.right));
    }
}

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

/**
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 cnt=0,nextcnt=1,depth=0;
        Queue<TreeNode> q = new LinkedList<TreeNode>();//建立一个队列
        q.offer(root);//先将根元素放入队列
        while(q.size()!=0){//如果队列不为0，则一直循环
            TreeNode tree = q.poll();//将一个元素出队列
            cnt++;//队列中每层的元素，每个元素出一次队列，计数值加1，对每层的元素个数进行计数
            if(tree.left!=null)//如果该元素的左子树不为空，则入队列
                q.offer(tree.left);
            if(tree.right!=null)
                q.offer(tree.right);
               if(cnt==nextcnt){//这里是判断队列中每一层的元素是否已全部出队列，若相等，则
                   nextcnt = q.size();//将nextsize的值重新设置为该层的节点数目，
                   cnt=0;//将计数值清0
                   depth++;//每执行一次，则说明该层的元素已全部出队列
               }
        }
        return depth;
    }
}