题解 | #第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); } } } Collections.sort(res); return res.get(k - 1); } }
代码使用的编程语言是Java。
该题考察的知识点是二叉树的遍历和数据结构的操作。
这段代码实现了一个方法kthLighest
,该方法接收一个树的根节点root
和一个整数k
作为参数,返回二叉树中第k大的节点的值。
代码的逻辑如下:
- 创建一个队列
queue
和一个列表res
,用于存储遍历过程中的节点和节点值。 - 如果根节点
root
不为空,将其入队列。 - 使用广度优先搜索(BFS)遍历二叉树:获取当前队列中的节点个数size。循环遍历队列中的节点:弹出队列中的节点cur。将节点cur的值添加到列表res中。如果节点cur的左子节点不为空,将其加入队列。如果节点cur的右子节点不为空,将其加入队列。
- 对列表
res
进行排序。 - 返回列表
res
中第k-1
个元素,即第k大的节点的值。