编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
示例 1:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 3
输出: true
示例 2:
输入:
matrix = [
[1, 3, 5, 7],
[10, 11, 16, 20],
[23, 30, 34, 50]
]
target = 13
输出: false
题解
一个很简单的二分,愣生生做错了好几次
先用了两次二分,虽然过了,但是复杂度有点高,如下
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
//行数
int rows=matrix.length;
if(rows==0) return false;
//列数
int columns=matrix[0].length;
if(columns==0){
return false;
}
int i=0,j=rows-1;
while(j>i){
int mid=(i+j+1)/2;
System.out.println(mid);
if(matrix[mid][0]>target){
j=mid-1;
}else if(matrix[mid][0]<target){
i=mid;
}else{
return true;
}
}
int use=i;
i=0;
j=columns-1;
while(j>i){
int mid=(i+j+1)/2;
if(matrix[use][mid]>target){
j=mid-1;
}else if(matrix[use][mid]<target){
i=mid;
}else{
return true;
}
}
if (matrix[use][j]==target){
return true;
}
return false;
}
}
随后发现用一次就够了…(真的是快)
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
if(matrix==null || matrix.length==0) return false;
int row=0,col=matrix[0].length-1;
while(row<matrix.length && col>=0){
if(target==matrix[row][col]){
return true;
}else if(target>matrix[row][col]){
row++;
}else{
col--;
}
}
return false;
}
}