二叉树基本的广度优先算法是借助队列的先入先出特点实现二叉树的广度优先算法,即一层一层的遍历二叉树,因此广度优先也是层序遍历,但是求取二叉树的深度,需要将二叉树的广度优先算法做一些修改,普通的广度优先算法是队列元素出队后就直接将当前节点元素的左右节点加入队列的尾部,而求深度则需要将一层的元素都d出才能算是一层深度,因此借助一个临时集合保存当前层所有d出的元素,d出完之后在将集合中元素的子节点加入队列继续进行,java实现:
public int maxDepth(TreeNode root) {
if (root == null) {
return 0;
}
Queue<TreeNode> nodeQueue=new LinkedList<>(); //广度优先队列
nodeQueue.offer(root); //根节点入队列
int depth=0;
while (!nodeQueue.isEmpty()){
List<TreeNode> pollNodeList=new ArrayList<>(); //临时list,保存当前层的所有节点,便于后续一次性加入当前层节点的子节点
while (!nodeQueue.isEmpty()){
pollNodeList.add(nodeQueue.poll()); //循环d出当前层的所有节点,并加入临时list
}
depth++; //每一层深度+1
pollNodeList.forEach(treeNode -> { //将当前层的所有节点的子节点加入队列
if (treeNode.left!=null){
nodeQueue.offer(treeNode.left);
}
if (treeNode.right!=null){
nodeQueue.offer(treeNode.right);
}
});
}
return depth; //返回深度
}
时间复杂度:O(n),其中 n 为二叉树的节点个数,每个节点均访问一次。
空间复杂度:此方法空间的消耗取决于队列存储的元素数量,其在最坏情况下会达到 O(n)。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)