今天和大家聊的问题叫做 有序矩阵中第 K 小的元素，我们先来看题面：https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix/
Given an n x n matrix where each of the rows and columns are sorted in ascending order, return the kth smallest element in the matrix.
Note that it is the kth smallest element in the sorted order, not the kth distinct element.
给你一个 n x n 矩阵 matrix ，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。请注意，它是 排序后 的第 k 小元素，而不是第 k 个 不同 的元素。
示例
示例 1： 输入：matrix = [[1,5,9],[10,11,13],[12,13,15]], k = 8 输出：13 解释：矩阵中的元素为 [1,5,9,10,11,12,13,13,15]，第 8 小元素是 13 示例 2： 输入：matrix = [[-5]], k = 1 输出：-5
解题
1）利用优先队列进行处理
class Solution { public: int kthSmallest(vector<vector<int>>& matrix, int k) { priority_queue<int> pq; for(int i=0;i<matrix.size();i++){ for(int j=0;j<matrix[0].size();j++){ pq.emplace(matrix[i][j]); if(pq.size()>k) pq.pop(); } } return pq.top(); } };
2）利用二分思路进行处理
class Solution { public: int kthSmallest(vector<vector<int>>& matrix, int k) { int left=matrix[0][0],right=matrix.back().back(); while(left<right){ int mid=left+(right-left)/2,cnt=0; for(int i=0;i<matrix.size();i++){ cnt+=upper_bound(matrix[i].begin(),matrix[i].end(),mid)-matrix[i].begin(); } if(cnt<k) left=mid+1; else right=mid; } return left; } };
好了，今天的文章就到这里，如果觉得有所收获，请顺手点个在看或者转发吧，你们的支持是我最大的动力 。
