378. Kth Smallest Element in a Sorted Matrix

Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix.

Note that it is the kth smallest element in the sorted order, not the kth distinct element.

Example:

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

return 13.

 

Note: 
You may assume k is always valid, 1 ≤ k ≤ n2.

题意：给出一个n维矩阵，矩阵中每行，每列元素都是递增的。要求求出矩阵中第k小的元素。

思路一： 先map映射保存每个元素出现次数，再从小到大查找元素。

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        map<int,int>mp;
        for(int i=0;i<matrix.size();i++){
            for(int j=0;j<matrix.size();j++){
                mp[matrix[i][j]]++;
            }
        }
        int cnt=0;
        for(auto it:mp){
            cnt+=it.second;
            if(cnt>=k)
            return it.first;
        }
        return 0; 
    }
}; 

思路二： 使用优先队列，将所有元素都入队，然后出队找到第k小元素。

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        priority_queue<int,vector<int>,greater<int> >que;
        for(int i=0;i<matrix.size();i++){
            for(int j=0;j<matrix.size();j++){
                que.push(matrix[i][j]);
            }
        }
        int cnt=1;
        while(cnt!=k){
            que.pop();
            cnt++;
        }
        return que.top();
    }
}; 

思路三：使用了折半查找的思想，这里对元素的值进行折半，而不是下标。

// minn+maxx可能超int
typedef long long ll;
class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        int n=matrix.size();
        // 运用折半查找的思想求解 初始取最大，最小值作为两个端点
        ll minn=matrix[0][0],maxx=matrix[n-1][n-1],mid;
        int cnt=0;
        while(minn<=maxx){
            mid=(minn+maxx)/2;
            int cnt=0;
            // upper_bound(begin,end,num)-begin 在[begin,end)区间(有序区间)
            //二分查找大于num的第一个数的下标  
            //lower_bound(begin,end,num)查找第一个大于等于num的数的下标
             for(int i=0;i<n;i++){
                // 即为寻找matrix数组第i行中小于mid的数的个数
                cnt+=upper_bound(matrix[i].begin(),matrix[i].end(),mid)-matrix[i].begin();
            }
            //cnt=k时也要继续向下寻找，因为可能此时mid并不存在于数组中
            if(cnt>=k){
                maxx=mid-1;
            }else {
                minn=mid+1;
            }
        }
        return minn;
    }
};