题解 | #牛群的最短路径#
牛群的最短路径
https://www.nowcoder.com/practice/c07472106bfe430b8e2f55125d817358
知识点:二叉树,层序遍历
题目要求找到从根节点到叶子节点的最短距离,也就是距离根节点最近的叶子节点,我们可以从根节点开始进行广度优先搜索,同时,每遍历到一层时进行计数,记录下来当前走过的层数,直至找到一个叶子节点,此时该叶子节点就是距离根节点最近的叶子节点,返回当前层数即为答案。
Java题解如下
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型
*/
public int minDepth (TreeNode root) {
// write code here
if(root == null) {
return 0;
}
int cnt = 0;
Queue<TreeNode> queue = new ArrayDeque<>();
queue.offer(root);
while(!queue.isEmpty()) {
int size = queue.size();
cnt++;
for(int i = 0; i < size; i++) {
TreeNode poll = queue.poll();
if(poll.left == null && poll.right == null) {
return cnt;
}
if(poll.left != null) {
queue.offer(poll.left);
}
if(poll.right != null) {
queue.offer(poll.right);
}
}
}
return cnt;
}
}
