[剑指offer-1384] 二维数组中的查找

题目描述：

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

输入：

输入可能包含多个测试样例，对于每个测试案例，
输入的第一行为两个整数m和n(1<=m,n<=1000)：代表将要输入的矩阵的行数和列数。
输入的第二行包括一个整数t(1<=t<=1000000)：代表要查找的数字。
接下来的m行，每行有n个数，代表题目所给出的m行n列的矩阵(矩阵如题目描述所示，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

输出：

对应每个测试案例，
输出”Yes”代表在二维数组中找到了数字t。
输出”No”代表在二维数组中没有找到数字t。

样例输入：

3 3
5
1 2 3
4 5 6
7 8 9
3 3
1
2 3 4
5 6 7
8 9 10
3 3
12
2 3 4
5 6 7
8 9 10

样例输出：

Yes
No
No

C 代码

#include<stdio.h>  
#include<stdbool.h>  
bool findNum(int* arr , int m , int n ,int key){
    if(arr == NULL || m<1 || n<1)
        return false;
    int row = 0;
    int col = n-1;
    while(row <= m-1 && col >= 0){
        if(arr[row*n+col] == key)
            return true;
        else if(arr[row*n+col] < key){
            row++;
        }else
            col--;
    }
    return false;
}
int main( ) {
    int m ,n;
    while(scanf("%d %d",&m,&n)!=EOF){
        int key;
        static int matrix[1000*1000] = {0};
        scanf("%d",&key);
        for(int i=0;i<m*n;i++)
            scanf("%d",matrix+i);
        bool result = findNum(matrix,m,n,key);
        if(result)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}