剑指offer系列源码--二维数组中的查找

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：14827

解决：2888

题目描述：

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

输入：

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

输入的第一行为两个整数m和n(1<=m,n<=1000)：代表将要输入的矩阵的行数和列数。

输入的第二行包括一个整数t(1<=t<=1000000)：代表要查找的数字。

接下来的m行，每行有n个数，代表题目所给出的m行n列的矩阵(矩阵如题目描述所示，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

输出：

对应每个测试案例，

输出”Yes”代表在二维数组中找到了数字t。

输出”No”代表在二维数组中没有找到数字t。

样例输入：

3 351 2 34 5 67 8 93 312 3 45 6 78 9 103 3122 3 45 6 78 9 10

样例输出：

YesNoNo

#include<iostream>
#include<stdio.h>
using namespace std;
bool findNum(int* arr,int rows,int colums,int number){

    if(arr==NULL||rows<=0||colums<=0){
        return false;
    }
    int row = 0,colum = colums-1;
    while(row<rows&&colum>=0){
        if(arr[row*colums+colum]==number){
            return true;
        }else if(arr[row*colums+colum]>number){
            --colum;
        }else{
            ++row;
        }
    }
    return false;
}
int main(){
    int rows,colums,number;
    while(scanf("%d%d",&rows,&colums)!=EOF){
        scanf("%d",&number);
        int *arr = new int[rows*colums];
        for(int i=0;i<rows*colums;i++){
            scanf("%d",&arr[i]);
        }
        if(findNum(arr,rows,colums,number)){
            printf("%s\n","Yes");
        }else{
            printf("%s\n","No");
        }
    }
    return 0;

}