题目3:二维数组中的查找(leetcode链接:https://leetcode-cn.com/problems/er-wei-shu-zu-zhong-de-cha-zhao-lcof/)
题目解析
题目分析
已知条件:二维数组每一行从左到右递增,每一列从上到下递增。
如果暴力遍历而数组,时间复杂度为O(m*n),而我们所直到的二分查找时间复杂度为O(logn);而二分法的思想是每次缩小目标值所在的范围,条件是目标数组有序;本题中目标数组是一个二维数组,虽然每行每列都是有序的,但是整体无序,故不能使用二分查找;要想降低二维数组的查找的时间复杂度,可以借助二分查找的思想,逐次缩小查找的范围,具体思路如下:
总结上述查找的过程,我们发现如下规律:首先选取数组中右上角的数字。如果该数字等于要查找的数字,查找过程结束;如果该数字大于要查找的数字,剔除这个数字所在的列;如果该数字小于要查找的数字,剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角,则每一次都在数组的查找范围中剔除一行或者一列,这样每一步都可以缩小查找的范围,直到找到要查找的数字,或者查找范围为空。
在前面的示例中,我们选取的是右上角,实际我们也可以选左下角。
代码描述
leetcode代码:
bool findNumberIn2DArray(int** matrix, int matrixSize, int* matrixColSize, int target){
//判断当前数组是否为空
if(matrixSize == 0)
return false;
int row = 0,col = matrixColSize[0] - 1;//行号、列号;
//控制行号列号不会超出范围
while(row < matrixSize && col >= 0)
{
//如果当前值小于目标值,如果当前行不是最后一行(row<matrixSize-1)则行号加1
if(matrix[row][col] < target && row < matrixSize-1)
row++;
else if( matrix[row][col] > target && col > 0)
col--;
else if(matrix[row][col] == target)
return true;
//如果行号大于等于col-1,列号小于等于0,直接退出
else
break;
}
return false;
}
官方代码:
bool Find(int* matrix, int rows, int columns, int number)
{
bool found = false;
if (matrix != NULL && rows > 0 && columns > 0)
{
int row = 0;
int column = columns - 1;
while (row < rows &&column >= 0)
{
if (matrix[row * columns + column] == number)
{
found = true;
break;
}
else if (matrix[row * columns + column] > number)
--column;
else
++row;
}
}
return found;
}