剑指offer—二维数组的查找
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数
博客中代码均在牛客C++11(clang++ 3.9)中通过
这里提供两种思路:
第一种是把每一行作为一个有序数组,通过二分查找来定位,找到返回true,找不到就到下一行继续查找,直到全部遍历查找后没有则返回false
代码如下:
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
//如果数组大小为空,返回false
if(0 == array.size())
return false;
//循环遍历每一行
for(int i=0;i<array.size();i++)
{
int left=0;
int right=array[i].size()-1;
//二分查找法查找target
while(left<=right)
{
int mid=left+((right-left)>>1);
if(target>array[i][mid])
left=mid+1;
else if(target<array[i][mid])
right=mid-1;
else
return true;
}
}
return false;
}
};
第二种是利用题目所给二维数组的特点:每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序
则此数组可写成矩阵形式,我们选取矩阵右上角或左下角元素与该整数进行比较,因为这两个位置向左(向上)和向下(向右)分别是递减和递增的,正好可以用来查找
这里选择右上角来说明:
如果整数大,说明整行元素都比该整数小,则向下偏移一行,
如果整数小,说明整列元素都比该整数大,则列向左偏移一列,
找到了则返回true,
当行或列走到边界仍没有找到,说明数组中没有该整数,则返回false
代码如下:
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
//如果数组大小为空,返回false
if(0 == array.size())
return false;
int row = 0;//用来标识每一行的下标
int col = array[0].size()-1;//用来标识每一列的下标
//以行列不越界作为循环条件,通过对行++和列--操作定位target
while(row<=array.size()-1 && col>=0)
{
if(target == array[row][col])//找到了,返回true
return true;
else if(target>array[row][col])//target比当前行最后一列大,则向下一行找
++row;
else//target比当前行最后一列小,则向前一列找
--col;
}
return false;//找不到,返回false
}
};
注意:这里针对这个题目给出这两种方法,如果题目有所变化就需要适当调整或者选择其他更好的方法,比如:只有行有序,数组每一行不等长的情况,第二种方法就不适用了