- 搜索二维矩阵
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
输出:true
示例 2:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
输出:false
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 100
-104 <= matrix[i][j], target <= 104
解法一:两次二分搜索
- 第一次二分:从第 0 列中的「所有行」开始找,找到合适的行
row
- 第二次二分:从
row
中「所有列」开始找,找到合适的列col
public boolean searchMatrix(int[][] matrix, int target) {
int low=0,high=matrix.length-1;
// 第一次二分:定位到所在行(从上往下,找到最后一个满足 matrix[x]][0] <= target的行号)
while(low<high){
//如果不加一会出现死循环
//比如low = 2, high = 3的时候,如果不加1,在满足low = mid的情况下,会一直死循环
int mid = low+high+1 >>1;
if(matrix[mid][0]<=target){
low=mid;
}else {
high=mid-1;
}
}
int row=high;
if(matrix[row][0]==target) return true;
if(matrix[row][0]>target) return false;
low=0;
high=matrix[0].length-1;
// 第二次二分:从所在行中定位到列(从左到右,找到最后一个满足 matrix[row][x] <= target的列号)
while(low<high){
int mid =low+high+1 >> 1 ;
if(matrix[row][mid]<=target){
low=mid;
}else {
high=mid-1;
}
}
int col=high;
return matrix[row][col]==target;
}
解法二:抽象 BST 解法
此文章创于本人学习时的记录,如有错误或更优解还请指出