题目描述
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。
请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
我想的不太快,就是先在第一行找target在哪一列,然后在这一列中寻找target
时间复杂度 $O(n+m)$
C++ 代码
class Solution {
public:
bool searchArray(vector<vector<int>> array, int target) {
if(array.empty())return false;
int n=array.size(),m=array[0].size();
int i,j,mid,k=0;
//cout<<n<<' '<<m<<endl;
if(array[0][0]>target)return false;
while(array[0][k]<target&&k<m)k++;
if(array[0][k]==target)return true;
k--;
//cout<<k<<endl;
i=0;j=n-1;
while(i<j){
mid=(i+j)/2;
if(array[mid][k]<target)i=mid+1;
else if(array[mid][k]>target)j=mid;
else return true;
//cout<<mid<<endl;
}
if(array[i][k]==target)return true;
return false;
}
};