今天在做二分查找标签的题的时候,发现题解看的不是很懂,就是这个题:
有序矩阵中第K小的元素,
给你一个 n x n 矩阵 matrix ,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。
请注意,它是 排序后 的第 k 小元素,而不是第 k 个 不同 的元素。
你必须找到一个内存复杂度优于 O(n2) 的解决方案。
示例 1:
输入:matrix = [[1,5,9],[10,11,13],[12,13,15]], k = 8
输出:13
解释:矩阵中的元素为 [1,5,9,10,11,12,13,13,15],第 8 小元素是 13
示例 2:
输入:matrix = [[-5]], k = 1
输出:-5
来源:力扣(LeetCode)
看了半天题解也没看懂,然后又debug看看执行情况,然后还是不懂,最后我就发现了我自己其实对二分查找这个算法理解的并不透彻,只是会写,然后差不多能用这个样子,然后就像这样,用法一变化,自己就不太好理解,所以以后对于算法还是完全理解每一步,每一个控制语句的作用,这样才能更好的掌握这个算法,以后用起来才会更加方便。
这是题解:
class Solution {
public int kthSmallest(int[][] matrix, int k) {
int n = matrix.length;
int left = matrix[0][0];
int right = matrix[n - 1][n - 1];
while (left < right) {
int mid = left + ((right - left) >> 1);
if (check(matrix, mid, k, n)) {
right = mid;
} else {
left = mid + 1;
}
}
return left;
}
public boolean check(int[][] matrix, int mid, int k, int n) {
int i = n - 1;
int j = 0;
int num = 0;
while (i >= 0 && j < n) {
if (matrix[i][j] <= mid) {
num += i + 1;
j++;
} else {
i--;
}
}
return num >= k;
}
}
来源:力扣(LeetCode)
首先再来理解一下二分查找:
最终的临界情况一定是
1.直接等于mid 如1,2,3求2
2.等于边界1,2中求1或2
其中left=mid+1;这句是一定不能变为left=mid;的,因为mid=(left+right)/2,这就导致了如果所求值为右边界那么会陷入死循环,如1,2,3求3,如果是left=mid那么left就会一直等于1,right一直等于2,从而死循环,而right却可以改为等于mid,因为mid=(left+right)/2,这就导致了mid本身的退一位,如果right=mid,本身就可以退一位而不会引起死循环;while(left<=right)这其中的=是控制最终的结果,可以根据自己的需求,来确定是否要加=。
本题中就有很大的改变,因为本身左右边界都不会超出所求值,所以最后left=right就是跳出循环的条件,此时的left就是所求的值,本题中用这样的方式来确定最后跳出循环的值一定在数组中。