230. 二叉搜索树中第K小的元素

给定一个二叉搜索树的根节点 root ,和一个整数 k ,请你设计一个算法查找其中第 k 个最小元素(从 1 开始计数)。

示例 1:


输入:root = [3,1,4,null,2], k = 1
输出:1
示例 2:


输入:root = [5,3,6,2,4,null,null,1], k = 3
输出:3
 

提示:

树中的节点数为 n 。
1 <= k <= n <= 104
0 <= Node.val <= 104

 public int kthSmallest(TreeNode root, int k) {
        PriorityQueue queue = new PriorityQueue();
        procrss(root, queue);
        int res = 0;
        for (int i = 0; i < k; i++) {
            res = queue.poll();
        }
        return res;
    }

    public void procrss(TreeNode root, PriorityQueue queue) {
        if (root == null) {
            return;
        }
        queue.add(root.val);
        procrss(root.left, queue);
        procrss(root.right, queue);
    }

你可能感兴趣的:(Leetcode,java,算法,leetcode,数据结构)