二维有序数组查找一个数

#include <iostream>
#include <cstdio>
using namespace std;


string search(int a[][1000], int m, int n, int key)
{
    int i = 0;
    int j = n - 1;
    while(i < m && j >= 0)
    {
        if (a[i][j] == key)
            return "Yes";
        else if (a[i][j] < key)
            i++;
        else
            j--;
    }


    return "No";
}


int main()
{
    int n, m;
    int a[1000][1000];
    while(scanf("%d%d", &m, &n) != EOF)
    {
        int key;
        scanf("%d", &key);
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
                scanf("%d", &a[i][j]);
        cout << search(a, m, n, key) << endl;
    }

}

样例输入：

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