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

题目描述:

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

 

输入:

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

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

 1 #include <iostream>
 2 #include <cstdio>
 3 using namespace std;
 4 
 5 string search(int a[][1000], int m, int n, int key)
 6 {
 7     int i = 0;
 8     int j = n - 1;
 9     while(i < m && j >= 0)
10     {
11         if (a[i][j] == key)
12             return "Yes";
13         else if (a[i][j] < key)
14             i++;
15         else
16             j--;
17     }
18 
19     return "No";
20 }
21 
22 int main()
23 {
24     int n, m;
25     int a[1000][1000];
26     while(scanf("%d%d", &m, &n) != EOF)
27     {
28         int key;
29         scanf("%d", &key);
30         for(int i = 0; i < m; i++)
31             for(int j = 0; j < n; j++)
32                 scanf("%d", &a[i][j]);
33         cout << search(a, m, n, key) << endl;
34     }
35 }

 

posted @ 2012-11-21 11:07  chkkch  阅读(4969)  评论(0编辑  收藏  举报