标签：.com 一个 ted 参考 return ace rip mat tps
给定一个 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
378 Kth Smallest Element in a Sorted Matrix 有序矩阵中第K小的元素
标签：.com 一个 ted 参考 return ace rip mat tps
原文地址：https://www.cnblogs.com/xidian2014/p/8848625.html
