题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
如下面的二维数组就是每行、每列都递增排序。
1 2 8 9
2 4 9 12
4 7 10 13
6 8 11 15
按照剑指offer上面作者的解答,有如下规律:
首先选取数组右上角的数字。如果该数字等于要查找的数字,查找过程结束;如果该数字大于要查找的数字,提出这个数字所在的列;如果该数字小于要查找的数字,提出这个数字所在的行。也就是说,如果要查找的数字不在数组的右上角,则每一次都在数组的查找范围中剔除一行或一列。同理,也可以选择左下角。
代码如下:
bool find_(int *matrix, int rows, int cols, int number)
{
bool flag = false;
if(matrix != NULL && rows > 0 && cols > 0)
{
int row = 0;
int col = cols - 1;
while(row < rows && col >= 0)
{
int val = matrix[row * cols + col];
if(val == number)
{
flag = true;
break;
}
else if(val > number)
col--;
else
row++;
}
}
return flag;
}
第二种思路就是二分。
首先以最小的行上二分列,可以剔除掉所有比要查找数字大的列;然后在最大的列上二分行,可以剔除掉所有比要查找数字小的行。
代码如下:
bool find_bin(int *matrix, int rows, int cols, int number)
{
if(matrix != NULL && rows > 0 && cols > 0)
{
int row = 0;
int col = cols - 1;
while(true)
{
int mid;
int l = 0;
int r = col;
while(l <= r)
{
mid = (int)(l + r) / 2;
if(matrix[row * cols + mid] == number)
{
return true;
}
else if(matrix[row * cols + mid] < number)
l = mid + 1;
else
r = mid -1;
}
col = l - 1;
if(col < 0)
break;
l = row;
r = rows - 1;
while(l <= r)
{
mid = (int)(l + r) / 2;
if(matrix[mid * cols + col] == number)
{
return true;
}
else if(matrix[mid * cols + col] < number)
l = mid + 1;
else
r = mid -1;
}
row = l;
if(row >= rows)
break;
}
}
return false;
}
下面是测试代码:
int main()
{
int matrix[] = {1, 2, 8, 9, 2, 4, 9, 12, 4, 7, 10, 13, 6, 8, 11, 15};
cout<<find_(matrix, 4, 4, 7)<<endl;
cout<<find_bin(matrix, 4, 4, 7)<<endl;
return 0;
}