题意
编写一个高效的算法来判断 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
思路
使用两个二分查找
第一个二分查找找到所在的列
第二个在列中找元素
代码演示
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
int m=matrix.length;
int n=matrix[0].length;
if(target<matrix[0][0]||target>matrix[m-1][n-1])
return false;
int findm = findM(matrix, target);
if(matrix[findm][0]==target)
return true;
int pre=0,end=n-1,mid;
while (pre<=end)
{
mid=(pre+end)/2;
if(matrix[findm][mid]==target)
return true;
else if(matrix[findm][mid] <target )
pre=mid+1;
else
end=mid-1;
}
return false;
}
//在第一列中进行二分查找
public int findM(int[][] matrix, int target)
{
int pre=0;
int end=matrix.length-1;
int mid;
//二分查找的目的是找出 大于这一行首元素,小于下一行首元素
while (pre<=end)
{
mid=(pre+end)/2;
if((mid+1<=matrix.length-1 &&matrix[mid][0] <=target && matrix[mid+1][0]>target)||mid==matrix.length-1)
return mid;
else if(matrix[mid][0]>target)
end=mid-1;
else
pre=mid+1;
}
return 0;
}
}
效果
info
解答成功:
执行耗时:0 ms,击败了100.00% 的Java用户
内存消耗:37.9 MB,击败了61.25% 的Java用户