题目描述:
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
分析:
由于二维数组是排好序的,一定是要采用和二分查找相似的思想,一维数组是找中间数,二维数组也是找中间数。而中间数就是逆对角线上的数!(左下角,或是右上角)。
这样一次就能排除一行或是一列,对于n*n的矩阵,复杂度O(n)。
1 | 2 | 3 | 4 | 5 | 6 |
4 | 5 | 5 | 7 | 9 | 12 |
5 | 6 | 6 | 8 | 12 | 13 |
6 | 7 | 7 | 9 | 12 | 16 |
代码实现:
bool Find(int target, vector<vector<int> > array)
{
int rows = array.size();
if(rows<=0)
return false;
int cols = array[0].size();
int mid_i= rows-1,mid_j = 0 ; //从左下角坐标开始查找
while( mid_i >=0 && mid_j <= cols-1){
if(array [mid_i][mid_j] == target){//查找成功
return true;
}
else if(array [mid_i][mid_j] < target){//数较小,排除这一列,列+1
mid_j++;
}
else{
mid_i--;//数较大,行-1
}
}
return false;
}
{
int rows = array.size();
if(rows<=0)
return false;
int cols = array[0].size();
int mid_i= rows-1,mid_j = 0 ; //从左下角坐标开始查找
while( mid_i >=0 && mid_j <= cols-1){
if(array [mid_i][mid_j] == target){//查找成功
return true;
}
else if(array [mid_i][mid_j] < target){//数较小,排除这一列,列+1
mid_j++;
}
else{
mid_i--;//数较大,行-1
}
}
return false;
}