关于二维有序数组中搜索,其题目如下:
给定一个矩阵,都是整数,每一行从左到右升序,每一列从上到下有序,例如下面的矩阵
[ [1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 50] ]请用最快的时间找出特定的数,例如,输入3,存在这个数,输入15,不存在这个数。
实现如下的函数bool FindArray(int *pArray,int nWidth,int nheight,int nKey)。
最笨的方法是逐个遍历进行比较,如果是这样,这题目就没什么意义了。观察这个数组,可以先从第一行的最后一个元素开始,这个元素所在的列,在7这个元素下面的都比他大,这个元素所在的列,在他左边的都比他小。那么我们就可以这样比较,首先,取出第一行最右边的元素,和关键字比较,如果关键字正好和他相等,那么就找到了,如果这个元素比关键字小,那么要查找的关键字必然在这一行的下面或者不存在,所以,就定位到这个元素的下面一行对应的元素,即第二行最后一列;如果这个元素比关键字大,那么关键字必然在这个元素所在列的左边,那么就定位到这个元素的左边,即第一行倒数第二个元素,如此,搜索完整个数组就可以找到结果。既然思路已经清楚了,那么代码也就很容易写出来:
bool FindArray(int *pArray,int nWidth,int nheight,int nKey)
{
if (NULL == pArray || nWidth <= 0 || nheight <= 0)
{
return false;
}
//从右上角开始搜索
int nCol = nWidth - 1;
int nRow = 0;
bool bFound = false;
while (nCol >= 0 && nRow <= nheight-1)
{
if (pArray[nRow*nWidth+nCol] == nKey)
{
bFound = true;
break;
}
else if (pArray[nRow*nWidth+nCol] < nKey)
{
nRow ++;
}
else
{
nCol --;
}
}
return bFound;
}
其实,还有另外一种方法,就是最先比较第一列最后一行的要素,其比较规则和上面的类似,不再陈述,在此也给出代码:
bool FindArrayEx(int *pArray,int nWidth,int nheight,int nKey)
{
if (NULL == pArray || nWidth <= 0 || nheight <= 0)
{
return false;
}
//从坐下角开始搜索
int nCol = 0;
int nRow = nheight - 1;
bool bFound = false;
while (nCol < nWidth && nRow >= 0)
{
if (pArray[nRow*nWidth+nCol] == nKey)
{
bFound = true;
break;
}
else if (pArray[nRow*nWidth+nCol] < nKey)
{
nCol ++;
}
else
{
nRow --;
}
}
return bFound;
}
void main()
{
int a[] = {1,2,8,9,2,4,9,12,4,7,10,13,6,8,11,15};
int b[] = {1, 3, 5, 7, 10, 11, 16, 20, 23, 30, 34, 50};
printf("%d\n",FindArrayEx(b,4,3,3));
}
如果有更好的方法也请交流,此题目在leetcode的地址是https://oj.leetcode.com/problems/search-a-2d-matrix/