题目说明
给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。
请注意,它是排序后的第k小元素,而不是第k个元素。
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix
分析:采用优先对列的方法,一定要注意,一定要遍历完所有元素
int kthSmallest(vector<vector<int>>& matrix, int k)
{
int n = matrix.size();
if (n == 0)
{
return 0;
}
priority_queue<int> pq;
for (int i = 0; i < matrix.size(); i++)
{
for (int j = 0; j < matrix[i].size(); j++)
{
pq.push(matrix[i][j]);
if (pq.size() > k)
{
pq.pop(); //采用优先队列,一定要将矩阵中的所有元素遍历一遍
}
}
}
return pq.top();
}