题目描述:
给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。
请注意,它是排序后的第 k 小元素,而不是第 k 个不同的元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
思路:
二分法,由于从左到右,从上到下都是升序,所以当定义一个元素,会有一条分割线,左上部分的数都是小于该元素的,坐下都是大于该元素,利用这一规律,找到,数量等于k的分割点
class Solution {
public:
bool check(vector<vector<int>>&matrix,int mid,int k,int n){
int i=n-1;
int j=0;
int num=0;
while(i>=0&&j<=n-1){
if(matrix[i][j]<=mid){
j++;
num+=i+1;
}
else{
i--;
}
}
return num>=k;
}
int kthSmallest(vector<vector<int>>& matrix, int k) {
int n=matrix.size();
int left=matrix[0][0];
int right=matrix[n-1][n-1];
while(left<right){
int mid=left+(right-left)/2;
if(check(matrix,mid,k,n)){
right=mid;
}
else{
left=mid+1;
}
}
return left;
}
};