378. Kth Smallest Element in a Sorted Matrix

楼主: Rushia (みけねこ的鼻屎)   2022-08-02 23:18:02
https://leetcode.com/problems/kth-smallest-element-in-a-sorted-matrix/
给予一个二维阵列表示的Matrix
这个矩阵的行和列都是有序的
找出矩阵内第k小的数字
第一眼看到直觉是二分搜寻
但是我太烂想了半小时想不出来
只好直接用MinHeap暴力解了
我就烂 唉 渍鲨
class Solution {
public int kthSmallest(int[][] matrix, int k) {
PriorityQueue<Integer> heap = new PriorityQueue<>();
for (int[] row : matrix)
for (int j = 0; j < matrix[0].length; j++)
heap.offer(row[j]);
int ans = 0;
for(int i = 0;i < k;i++)
ans = heap.poll();
return ans;
}
}

Links booklink

Contact Us: admin [ a t ] ucptt.com