378. 有序矩阵中第K小的元素
给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。
请注意,它是排序后的第 k 小元素,而不是第 k 个不同的元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
提示:
你可以假设 k 的值永远是有效的,1 ≤ k ≤ n2 。
二分查找
class Solution {
public int kthSmallest(int[][] matrix, int k) {
int left = matrix[0][0];
int right = matrix[matrix.length-1][matrix.length-1];
while(left<right) {
int mid = ((left+right)>>1);
if(check(matrix,k,mid)) {
right = mid;
}else {
left = mid+1;
}
}
return left;
}
public boolean check(int[][] matrix,int k,int mid) {
int i = matrix.length-1;
int j = 0;
int numb =0;
while(i>=0&&j<matrix.length) {
if(matrix[i][j]<=mid) {
j++;
numb +=i+1;
}else {
i--;
}
}
return numb>=k;
}
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。