二叉树深度
二叉树中多数算法都是用递归实现的。
解题思路:
递归就是把大规模问题拆分成小规模问题,分而治之。
若求以3为根节点的深度,可以求以9、20 为根节点树深度的最大值加一。
原问题:求以3为根节点的树的深度
子问题:求以9为根节点的树的深度
子问题:求以20为根节点的树的深度
实现思路:
1.传递参数
约束子问题的输入与输出
输入根节点 输出深度
2.单层逻辑
约定第一次递归要干什么
单层二叉树的深度为max(左子树深度,右子树深度)+1
3.推出条件
递归何时结束
root为null 深度自然为0 return 0 退出
就比方说递归到9时,其左右子树都为空,所以返回值为max(0,0)+1也就是1,所以9为根的树深度为1.
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
import java.lang.Math;
public class Solution {
public int TreeDepth(TreeNode root) {
//退出条件
if(root==null){
return 0;
}
//单层逻辑
//计算左右子树的长度,并且比较获取其最大值加一为树的深度
// 传递的参数:输入根节点 输出深度int
int left =TreeDepth(root.left);
int right=TreeDepth(root.right);
// 传递的参数:输入根节点 输出深度int
return Math.max(left,right)+1;
}
}
利用BFS+队列实现
//使用队列+BFS解决问题
import java.util.*;
public class Solution {
public int TreeDepth(TreeNode root) {
//根节点为空直接退出取
if(root==null){
return 0;
}
int res=0;
//引入辅助队列实现BFS
Queue<TreeNode> queue=new LinkedList<>();
queue.offer(root); //压入根节点
//遍历每一层
while(!queue.isEmpty()){
//每遍历一层就让树的深度加一
res++;
//queue每次只保存当层节点数
int size=queue.size();
for(int i=0;i<size;i++){
//出队节点
TreeNode node=queue.poll();
//找该节点的邻接节点入队,针对树来说就是其左右子树
//而且不用考虑 环路问题 直接访问即可
if(node.left!=null){
queue.offer(node.left);
}
if(node.right!=null){
queue.offer(node.right);
}
}}
return res;
}
}