编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
执行用时:0 ms, 在所有 Java 提交中击败了100.00%的用户
链接:https://leetcode-cn.com/problems/search-a-2d-matrix
思路:先从矩阵的第一列开始,用一次二分查找,如果第一次行的mid值所对应的元素大于或小于要搜索的值,再对该行的所有列进行二分查找,如果找到返回true,没再该行找到,就更新行值,直到外部的while循环结束,仍没有找到则返回false;
代码:
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
int endRow = matrix.length-1;
int endCol = matrix[0].length-1;
int begin = 0;
int end = endRow;
while(begin<=end)
{ int mid = begin + ((end-begin)>>1);
if(matrix[mid][0]>target)
{ int begin1 = 0;
int end1 = endCol;
while(begin1<=end1)
{
int mid1 = begin1 +((end1-begin1)>>1);
if(matrix[mid][mid1]>target)
{
end1 = mid1 -1;
}
else if(matrix[mid][mid1]<target)
{
begin1 = mid1 + 1;
}
else{
return true;
}
}
end = mid -1;
}
else if(matrix[mid][0]<target)
{ int begin2 = 0;
int end2 = endCol;
while(begin2<=end2)
{ int mid2 = begin2 + ((end2-begin2)>>1);
if(matrix[mid][mid2]>target)
{
end2 = mid2 -1;
}
else if (matrix[mid][mid2]<target)
{
begin2 = mid2 +1;
}
else{
return true;
}
}
begin = mid +1;
}
else{
return true;
}
}
return false;
}
}