题解 | #二叉搜索树的第k个节点#使用小顶堆+递归解决
二叉搜索树的第k个节点
https://www.nowcoder.com/practice/57aa0bab91884a10b5136ca2c087f8ff
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 proot TreeNode类 * @param k int整型 * @return int整型 */ public int KthNode (TreeNode proot, int k) { int res = -1; //构建小顶堆,默认是小的值在最上面 Queue<Integer> queue = new PriorityQueue<>(); dfs(proot, queue); if (queue.isEmpty() || k > queue.size()) { return res; } //获取结果 for (int i = 0; i < k; i++) { res = queue.poll(); } return res; } //先序遍历,根->左->右,将结果存入到小顶堆中 public void dfs(TreeNode proot, Queue<Integer> queue) { if (proot == null) { return; } queue.offer(proot.val); dfs(proot.left, queue); dfs(proot.right, queue); } }