面试题3：二维数组中的查找

题目链接：http://ac.jobdu.com/problem.php?pid=1384

思路：每次将要查找的数字t与二维数组右上角的元素比较。

1、t比右上角的元素大，那么t肯定比该元素所在行的所有元素都大，直接删除该行，更新右上角元素。

2、t比右上角的元素小，那么t肯定比该元素所在列的所有元素都小，直接删除该列，更新右上角元素。

3、t等于右上角的元素，直接返回"Yes"。

code：

#include <cstdio>
using namespace std;
const int MAXN = 1005;
int A[MAXN][MAXN];
bool inArray(int m, int n, int t)
{
    int i = 0;
    int j = n - 1;
    while (i < m && j >= 0)
    {
        if (A[i][j] > t) --j;
        else if (A[i][j] < t) ++i;
        else return true;
    }
    return false;
}
 
int main()
{
    int m, n, t;    // m：行数 n：列数 t：待查找的数字
    while (scanf("%d %d %d", &m, &n, &t) != EOF)
    {
        // 输入数据
        for (int i = 0; i < m; ++i)
        {
            for (int j = 0; j < n; ++j)
            {
                scanf("%d", &A[i][j]);
            }
        }
        // 预判
        if (t < A[0][0] || t > A[m - 1][n - 1])
        {
            puts("No");
            continue;
        }
        if (inArray(m, n, t)) puts("Yes");
        else puts("No");
    }
    return 0;
}