[九度][何海涛] 二维数组中的查找

题目描述：

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

 

输入：

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

输入的第一行为两个整数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

这题在LeetCode上有相似的，具体的细节我在以后会写，可以做到O(n)

#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;
    }
}