LeetCode 378——有序矩阵中第K小的元素（优先队列）

一、题目介绍

给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第k小的元素。
请注意，它是排序后的第k小元素，而不是第k个元素。

示例:

matrix = [
   [ 1,  5,  9],
   [10, 11, 13],
   [12, 13, 15]
],
k = 8,

返回 13。
说明: 
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。

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

二、解题思路

本题只需维护一个长度为k的最大堆，既可以解决。每次插入数据，需要O(nlogn)的时间复杂度进行排序。详细请见代码

三、解题代码

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        priority_queue<int> que;  //最大堆
        int m = matrix.size();
        int n = matrix[0].size();
        int p = 0;
        for(int i = 0; i < m; ++i)
        {
            for(int j = 0; j < n; ++j)
            {
                if(que.empty())     //当优先队列为空
                {
                    que.push(matrix[i][j]);
                    p++;
                }
                else
                {
                    if(p < k)   //优先队列中元素的个数不大于k
                    {
                        que.push(matrix[i][j]);
                        p++;
                    }
                    else
                    {
                        if(que.top() > matrix[i][j])
                        {
                            que.pop();
                            que.push(matrix[i][j]);
                        }
                    }
                }
            }
        }
        return que.top();
    }
};

四、解题结果