在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
在一个有序数组中查找某个数,我们很容易就想到二分查找,所以不难写出代码:
public class Solution {
public boolean Find(int target, int [][] array) {
int row=array.length,col=array[0].length;
boolean flag=false;
for(int i=0;i<row;i++){
int r=0,l=col-1,mid=0;
while(r<=l){
mid=(r+l)/2;
if(array[i][mid]<target) r=mid+1;
else if(array[i][mid]>target) l=mid-1;
else return true;
}
}
return flag;
}
}
除了这种方法以外,还有其他方法吗?
我们试着分析一下题意,既然从左到右依次增大,从上到下也依次增大,那我们可不可以利用这一点呢?
这里有个小窍门,因为如果当前位置比target小,那我们到底是向下还是向右呢?你可能会说我们可以用bfs来解决呀,但是这样就把问题复杂话了。我们应该将这种双选题变为单选题!
我们选择将右上角或者左下角作为起点,以右上角为例:如果当前位置比target小,那么我们就往左;如果比当前位置大,那么我们就往下。这样这个问题就完美解决了!
右上角作为起点:
public class Solution {
public boolean Find(int target, int [][] array) {
int rowTol=array.length-1,col=array[0].length-1,row=0;
boolean flag=false;
while(col>=0 && row<=rowTol){
if(array[row][col]>target) col--;
else if(array[row][col]<target) row++;
else return true;
}
return flag;
}
}
左下角作为起点:
public class Solution {
public boolean Find(int target, int [][] array) {
int row=array.length-1,colTol=array[0].length-1,col=0;
boolean flag=false;
while(row>=0 && col<=colTol){
if(array[row][col]>target) row--;
else if(array[row][col]<target) col++;
else return true;
}
return flag;
}
}