二维数组中的查找

题目

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

思路

   首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数组，剔除这个数字所在的列；如果该数字小于要查找的数字，剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。

代码

#include <iostream>
#include <algorithm>

using namespace std;

bool Find(int* matrix, int rows, int columns, int number) {
    bool found = false;
    if(matrix != NULL && rows > 0 && columns > 0) {
        for(int row = 0, column = columns - 1;row < rows && column > 0;) {
            if(matrix[row * columns + column] == number) {
                found = true;
                break;
            }
            else if(matrix[row * columns + column] > number) {
                column--;
            }
            else {
                row++;
            }
        }
    }
    return found;
}

int main() {
    //测试用例1
    int test1[3][3] = {{1,4,6},{2,5,9},{4,6,10}};
    //测试用例2
    int* test2 = NULL;
    bool res1 = Find((int*)test1, 3, 3, 9);
    bool res2 = Find((int*)test1, 3, 3, 20);
    bool res3 = Find(test2, 3, 3, 9);
    string s1 = res1 ? "数组test1中存在数字9" : "数组test1中不存在数字9";
    string s2 = res2 ? "数组test1中存在数字20" : "数组test1中不存在数字20";
    string s3 = res3 ? "数组test2中存在数字9" : "数组test2中不存在数字9";
    cout<<s1<<endl;
    cout<<s2<<endl;;
    cout<<s3<<endl;
    return 0;
}