leetcode心得
#74. Search a 2D Matrix
- 用两次binary search
- 用一次binary search
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
//using two binary search
//check errors
if(matrix==null||matrix.length==0){
return false;
}
if(matrix[0]==null||matrix[0].length==0){
return false;
}
int row = matrix.length;
int col = matrix[0].length;
//通过每一行第一个元素确定在哪一行
int start=0,end=row-1;
while(start+1<end){
int mid=start+(end-start)/2;
if (matrix[mid][0]==target){
return true;
}
if(matrix[mid][0]>target){
end=mid;
}
else {
start=mid;
}
}
if(matrix[end][0]<=target){//end必须放在前面,因为start和end可能同时满足
row=end;//在这里需要找到目标所在的行
}
else if(matrix[start][0]<=target){ //如果这里改成if,就不正确
row=start;
}
else{
return false;
}
//else if(matrix[start][0]<target&&matrix[start+1][0]>target){ 直接顺序执行,因为找出了所在行,所以不用判断,判断的话会出现会出现越界
int hi=0, lo=col-1;
while(hi+1<lo){
int p = hi+(lo-hi)/2;
if (matrix[row][p]==target){
return true;}
if (matrix[row][p]>target){
lo=p;
}
else {
hi=p;
}
}
if (matrix[row][hi]==target){
return true;
}
else if (matrix[row][lo]==target){
return true;
}
return false;
}
}
用一次二分搜索
public boolean searchMatrix(int[][] matrix, int target) {
// write your code here
if(matrix == null || matrix.length == 0){
return false;
}
if(matrix[0] == null || matrix[0].length == 0){
return false;
}
int row = matrix.length;
int column = matrix[0].length;
int start = 0, end = row * column - 1;
while(start <= end){
int mid = start + (end - start) / 2;
int number = matrix[mid / column][mid % column];//矩阵的正中间点
if(number == target){
return true;
}else if(number > target){
end = mid - 1;
}else{
start = mid + 1;
}
}
return false;
}
}