[LeetCode][C++]有序矩阵中第K小的元素

有序矩阵中第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$ 。

---

思路：
遍历数组，维持一个大小为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();
    }
};

结果：