题解 | #牛群的最短路径#
牛群的最短路径
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; } }