378 Kth Smallest Element in a Sorted Matrix 有序矩阵中第K小的元素

Posted 2020-10-31 lina2014

给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第k小的元素。
请注意，它是排序后的第k小元素，而不是第k个元素。
示例:
matrix = [
   [ 1,  5,  9],
   [10, 11, 13],
   [12, 13, 15]
],
k = 8,
返回 13。
说明:
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。
详见：https://leetcode.com/problems/kth-smallest-element-in-a-sorted-matrix/description/

C++：

方法一：

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        priority_queue<int> q;
        for (int i = 0; i < matrix.size(); ++i)
        {
            for (int j = 0; j < matrix[i].size(); ++j) 
            {
                q.emplace(matrix[i][j]);
                if (q.size() > k) 
                {
                    q.pop();
                }
            }
        }
        return q.top();
    }
};

 方法二：

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;
    }
};

 参考：https://www.cnblogs.com/grandyang/p/5727892.html