1985. 找出数组中的第 K 大整数

题目

给你一个字符串数组 nums 和一个整数 k 。nums 中的每个字符串都表示一个不含前导零的整数。

返回 nums 中表示第 k 大整数的字符串。

注意：重复的数字在统计时会视为不同元素考虑。例如，如果 nums 是 [“1”,”2”,”2”]，那么 “2” 是最大的整数，”2” 是第二大的整数，”1” 是第三大的整数。

示例 1：

输入：nums = [“3”,”6”,”7”,”10”], k = 4
输出：”3”
解释：
nums 中的数字按非递减顺序排列为 [“3”,”6”,”7”,”10”]
其中第 4 大整数是 “3”

示例 2：

输入：nums = [“2”,”21”,”12”,”1”], k = 3
输出：”2”
解释：
nums 中的数字按非递减顺序排列为 [“1”,”2”,”12”,”21”]
其中第 3 大整数是 “2”

示例 3：

输入：nums = [“0”,”0”], k = 2
输出：”0”
解释：
nums 中的数字按非递减顺序排列为 [“0”,”0”]
其中第 2 大整数是 “0”

提示：

1 <= k <= nums.length <= 10^4
1 <= nums[i].length <= 100
nums[i] 仅由数字组成
nums[i] 不含任何前导零

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-the-kth-largest-integer-in-the-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

思路很简单，使用一个大小为的优先队列，实现上是一个小顶堆，一开始添加数组前个字符串，后面的字符串如果大于堆顶的就以一换一，最后堆顶的就是第大的。字符串比较的规则是，如果长度一样就从高位比较，如果不同则一定是长度长的更大。

通过这个题了解了PriorityQueue.comparator().compare()，可以获取的比较器中的方法，不用再重复写一遍比较函数了。

代码

class Solution {
    public String kthLargestNumber(String[] nums, int k) {
        PriorityQueue<String> minHeap = new PriorityQueue<>(k, (a, b) -> a.length() == b.length() ? a.compareTo(b) : a.length() - b.length());
        int n = nums.length;
        for (int i = 0; i < n; i++) {
            if (i < k) {
                minHeap.offer(nums[i]);
            } else if (minHeap.comparator().compare(nums[i], minHeap.peek()) > 0) {
                minHeap.poll();
                minHeap.offer(nums[i]);
            }
        }
        return minHeap.peek();
    }
}