题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
思路1:广度优先遍历(层次遍历)
使用队列存放结点,通过变量记录当前层结点数,下一层结点数,以及当前层已遍历结点数,以此来获知树的深度depth。
注意点:
虽然用的是队列ArrayDeque,且其包含pop(),push()方法,但它们遵照的依旧是栈思想。因此,要想使用队列的先进先出思想,可使用add()/addLast(),pollFirst()方法.
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.util.ArrayDeque;
public class Solution {
public int TreeDepth(TreeNode root) {
int depth=0,now=0,nextLayer=0,thisLayer=1;
ArrayDeque<TreeNode> q=new ArrayDeque<>();
if(root==null)return depth;
q.add(root);
while(!q.isEmpty()){
TreeNode temp=q.pollFirst();
now++;
if(temp.left!=null){
q.add(temp.left);
nextLayer++;
}
if(temp.right!=null){
q.add(temp.right);
nextLayer++;
}
if(now==thisLayer){
thisLayer=nextLayer;
now=0;
nextLayer=0;
depth++;
}
}
return depth;
}
}
思路2:递归
递归边界:结点的左/右结点为空
循环体:每个结点的深度是它的左子树与右子树深度的较大者+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) {
if(root==null)return 0;
int left=TreeDepth(root.left);
int right=TreeDepth(root.right);
return Math.max(left,right)+1;
}
}