码迷,mamicode.com
首页 > 其他好文 > 详细

有序二维数组中搜索特定的数

时间:2014-08-27 20:34:58      阅读:204      评论:0      收藏:0      [点我收藏+]

标签: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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!