面试题解：二维数组中的查找（C 语言版）

题目描述：

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

输入：

输入可能包含多个测试样例，对于每个测试案例，

输入的第一行为两个整数m和n(1<=m,n<=1000)：代表将要输入的矩阵的行数和列数。

输入的第二行包括一个整数t(1<=t<=1000000)：代表要查找的数字。

接下来的m行，每行有n个数，代表题目所给出的m行n列的矩阵(矩阵如题目描述所示，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

输出：

对应每个测试案例，

输出”Yes”代表在二维数组中找到了数字t。

输出”No”代表在二维数组中没有找到数字t。

 

解题思路：先定位所在行，然后二分查找。

/********************************

    Author: fduan

    Time:870 ms

    Memory:4816 kb

*********************************/

#include <stdio.h>
#include <stdlib.h>

int binary_search( int x[], int n, int t )
{
	int left = 0, right = n - 1, mid;
	while( left <= right )
	{
		mid = ( left + right ) / 2;
		if( x[mid] == t )
			break;
		else if( x[mid] < t )
			left = mid + 1;
		else
			right = mid - 1;
	}
	return left > right ? -1 : 1;
}

int search_elem( int x[], int m, int n, int t )
{
	int i, res = -1;
	for( i = 0; i < m; ++i )
	{
            // 目前的版本效率并不是最高的，后续博文将给出更高效的改进算法
		if( x[i * n] <= t && x[i * n + n - 1] >= t )
		{
			res = binary_search( x + i * n, n, t );
			if( res != -1 ) break;
		}
	}
	return res;
}

int main( int argc, char * argv[] )
{
	int m, n, t, i, j;
	int idx;
	static int x[1000*1000] = {0};
	while( scanf( "%d %d %d", &m, &n, &t ) != EOF ) 
	{
		for( i = 0; i < m * n; ++i )
			scanf( "%d", x + i );

		idx = search_elem( x, m, n, t );
		printf( idx == -1 ? "No\n" : "Yes\n" );
	}
	return 0;
}