给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。
请注意,它是排序后的第k小元素,而不是第k个元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
说明:
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。
-------------------------------------------------------------------------------------------------------------------------------
如果用STL中的库函数的话,那么可以写成
- 两个迭代器相减代表的是两个之间的距离。
- lower_bound() 返回一个迭代器,指向第一个不小于目标val的元素。
- upper_bound()返回一个迭代器,指向第一个大于目标val的元素。
class Solution {
public:
int h,w;
int count(vector<vector<int>>& matrix, int t){
int i = h - 1;
int j = 0,sum = 0;
while(i >= 0 && j <= w - 1){
if (matrix[i][j] <= t){
++j;
sum += i + 1;
}else{
--i;
}
}
return sum;
}
int kthSmallest(vector<vector<int>>& matrix, int k) {
h = matrix.size();
w = matrix[0].size();
int l = matrix[0][0];
int r = matrix[h - 1][w - 1];
while(l < r){ // 注意这里二分的写法,是求二分的左边界
int mid = l + r >> 1;
if (count(matrix, mid) >= k){
r = mid;
}else{
l = mid + 1;
}
}
return l;
}
};
问题一:
怎么确定函数返回的l就一定在矩阵中呢?
就拿题目中的例子来说,单独看13和14都满足count()结果为8,但是应该返回的是13。为什么呢? 因为如果14是存在于数组中的,13的count()就不可能是8,而应该是7,因为此时比13大的数有14、15。总结下:[13,15)之间的数是有着同一个count()的,是原因13在数组中,而(13,15)之间的数又不在数组中。
问题二:
count()函数的工作方式是什么样的?