编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-a-2d-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public static boolean searchMatrix(int[][] matrix, int target) {
if (matrix.length == 0 || matrix[0].length == 0)
return false;
int low = 0, high = matrix.length * matrix[0].length - 1, mid = 0;
int midval = 0;
while (low < high) {
mid = (low + high)/2;
midval = matrix[mid / matrix[0].length][mid % matrix[0].length];
if (midval < target)
low = mid + 1;
else
high = mid;
}
return matrix[low / matrix[0].length][low % matrix[0].length] == target;
}
}
将二维矩阵看作一维,使用二分查找,用除法和取余得到对应的二维下标。