剑指offer 二维数组中的查找

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

思路：我们都能想到O(n2)的算法。
但是有一个更好的O(n+m)的算法。根据题目给出的数组的性质我们可以推出一些规律，即：如果当前数字cur大于题目所给数字t，则cur所在行右边的元素以及下边元素都大于t，如果小于t，则cur所在行的左边的元素以及上边的元素都小于t，因此如果我们将开始的元素放在右上方，则如果t大于其，则直接向下走一行，如果小于其，则直接向左走一列。同理左下角也可以。

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

const int maxn = 1000 + 10;

int num[maxn][maxn];
int n, m;
int x;

bool find_(){
    int p1 = 0;
    int p2 = m - 1;
    while(p1 < n && p2 >= 0){
        if(num[p1][p2] > x){
            p2--;
        }
        else if(num[p1][p2] < x){
            p1++;
        }
        else{
            return true;
        }
    }
    return false;
}

int main()
{

    while(scanf("%d%d", &n, &m) != EOF){
        scanf("%d", &x);
        memset(num, 0, sizeof(num));
        for(int i = 0; i < n; i++)
            for(int j = 0; j < m; j++)
                scanf("%d", &num[i][j]);
        if(find_())
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}