题解 | #二叉搜索树的第k个节点#使用小顶堆+递归解决

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);
    }

}