题解 | #第k轻的牛牛# java
第k轻的牛牛
https://www.nowcoder.com/practice/d3b31f055b1640d9b10de0a6f2b8e6f3
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类 * @param k int整型 * @return int整型 */ public int kthLighest (TreeNode root, int k) { // write code here Queue<TreeNode> queue = new LinkedList<>(); List<Integer> res = new ArrayList<>(); if (root != null) { queue.offer(root); } while (!queue.isEmpty()) { int size = queue.size(); for (int i = 0; i < size; i++) { TreeNode cur = queue.poll(); res.add(cur.val); if (cur.left != null) { queue.offer(cur.left); } if (cur.right != null) { queue.offer(cur.right); } } } res.sort(null); return res.get(k - 1); } }
该代码使用的编程语言是Java
该题考察的知识点是二叉树的层序遍历和查找第k小的节点值。
代码的解释如下:
- 首先定义了一个
TreeNode
结构体,用于表示二叉树的节点。包括节点的值(val
)、左子节点(left
)和右子节点(right
)。 - 接着定义了一个
Solution
类,包含了一个kthLighest
方法,用于查找给定二叉树中第k小的节点值。 - 在
kthLighest
方法中,使用队列que
来进行层序遍历。将根节点加入队列,然后循环直到队列为空。 - 每次循环首先获取当前层的节点数
size
,然后在循环中取出当前层所有节点。 - 将当前节点的值加入结果列表
res
中,并将当前节点的左子节点和右子节点加入队列。 - 遍历完所有层后,对结果列表进行排序,然后返回第k个最小的节点值。
总结起来,该段代码实现了使用队列进行层序遍历,将二叉树所有节点值存储在列表中,并通过对列表进行排序获得第k小的节点值。