题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
代码一:用递归求解
public class Solution {
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;
}
}
代码二:用层次遍历解决
背景知识:
- 队列的初始化
import java.util.LinkedList;
import java.util.LinkedList;
Queue<TreeNode> queue =new LinkedList<>();
queue.add() 队列的加入
queue.poll() 队列的取出
queue.size() 队列的大小
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
public int TreeDepth(TreeNode root){
if (root==null)return 0;
Queue<TreeNode> queue=new LinkedList<TreeNode>();
queue.add(root);
int count=0,curcount=1,deep=0; //deep代表深度
//count 和 curcount用来判断弹出的节点是否为该层的节点,全部弹出后才能加入
while(!queue.isEmpty()){
TreeNode top=queue.poll(); //弹出节点
count++;
if(top.left!=null)queue.add(top.left); //加入弹出节点的左右子节点
if(top.right!=null)queue.add(top.right);
if (count==curcount){
deep++;
count=0;
curcount=queue.size(); //重置下次应弹出的数量,用于判断该层节点是否全部弹出
}
}
return deep;
}
}