题目:在一个二维数组中,每行都按照从左到右的递增的顺序排序,每列按照从上到下递增的顺序排序,完成一个函数,输入一个这样的二维数组和一个整数,判断数组中是否含有该整数。
思路:每次都选取数组中右上角的数字进行比较。如果该数字等于要查找的数字,则查找结束;如果该数字大于查找的数字,则剔除这个数字所在列;如果该数字小于查找的数字,则剔除这个数字所在行,这样每一步都可缩小查找范围,直到找到要查找的数字或者查找范围为空。
实现
bool findNumInMatrix(const vector<int>&TestData,int rows,int cols,int x)
{
bool found = false;
if (TestData.size() == 0 || rows <= 0 || cols <= 0) return found;
int row = 0;
int col = cols - 1;
while (row < rows && cols >= 0)
{
if (TestData[row * cols + col] == x)
{
found = true;
break;
}
else if(TestData[row * cols + col] > x)
{
col--;
}
else row++;
}
return found;
}
由以上思想可知,从矩阵的左下角开始查找也可实现相同功能,但从左上角或右下角开始则不可以。
从左下角开始的实现
bool findNumInMatrix(const vector<int>&TestData, int rows, int cols, int x)
{
bool found = false;
int row = rows-1;
int col = 0;
if (TestData.size() == 0 || rows <= 0 || cols <= 0) return found;
while (row >= 0 && col < cols)
{
if (TestData[row*col + col] < x)col++;
else if (TestData[row*col + col] > x) row--;
else
{
found = true;
break;
}
}
}