有序矩阵中第K小的元素
给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。
请注意,它是排序后的第 k 小元素,而不是第 k 个不同的元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
提示:
你可以假设 k 的值永远是有效的,1 ≤ k ≤
n
2
n^{2}
n2 。
思路:
遍历数组,维持一个大小为k的大顶堆。最后返回堆顶元素即可。这个方法好像并不在乎是不是有序矩阵。
代码:
class Solution {
public:
int kthSmallest(vector<vector<int>>& matrix, int k) {
priority_queue<int, vector<int>, less<int> > big_q;
int rows=matrix.size();
int cols=matrix[0].size();
int count=0;
for(int i=0;i<rows;i++)
{
for(int j=0;j<cols;j++)
{
if(count++<k)
{
big_q.push(matrix[i][j]);
}else{
if(matrix[i][j]<big_q.top())
{
big_q.pop();
big_q.push(matrix[i][j]);
}
}
}
}
return big_q.top();
}
};
结果: