leetcode378有序矩阵中第K小的元素[1]
给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。
请注意,它是排序后的第 k 小元素,而不是第 k 个不同的元素.
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
提示:
你可以假设 k 的值永远是有效的,1 ≤ k ≤ n2 。
因为是有序数组查找,首先想到的就是二分法。首先确定查找范围,low=matrix[0][0],high=matrix[n-1][n-1],第k小元素一定在[low, high]区间。每次对区间二分,查找<=mid的元素个数cnt,若cnt<k,则第k小元素一定在[mid+1, high]区间,否则在[low, mid]之间。这里需要注意的是:当cnt==k时不能认为mid就是第k小元素,因为mid不一定是矩阵中的数,需要对[low, mid]区间不断逼近,找到第k小元素。
按照上述思路,实现代码如下:
int kthSmallest(vector<vector<int>>& matrix, int k) {
int n = matrix.size();
int low = matrix[0][0], high = matrix[n-1][n-1];
w