标签:style http color os ar 代码 amp sp 时间
关于二维有序数组中搜索,其题目如下:
给定一个矩阵,都是整数,每一行从左到右升序,每一列从上到下有序,例如下面的矩阵
[ [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/
标签:style http color os ar 代码 amp sp 时间
原文地址:http://blog.csdn.net/zhouxuguang236/article/details/38875279