题目描述:
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
-
输入:
-
输入可能包含多个测试样例,对于每个测试案例,
输入的第一行为两个整数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;
}