038二叉树的深度
题目:
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
方法一:递归版本
思路:在返回的时候对左右子树的最大值+1
代码:
public int TreeDepth(TreeNode root) {
if(root==null) return 0;
int left =TreeDepth(root.left);
int right = TreeDepth(root.right);
return Math.max(left,right)+1;
}
方法二:非递归版本
思路:创建一个队列,先进先出,每次返回队列的首元素,如果有左右子树就入队列,利用前一层元素出队列后,队列只剩下一层元素,设置一个nextcount值,来实现层数的++;
代码:
public int TreeDepth(TreeNode root) {
if(root==null) return 0;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
int depth=0,count=0,nextCount=1;
while(queue.size()!=0){
TreeNode top = queue.poll();
count++;
if(top.left!=null){
queue.add(top.left);
}
if(top.right!=null){
queue.add(top.right);
}
if(count==nextCount){
depth++;
count=0;
nextCount=queue.size();
}
}
return depth;
}