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

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

当我们需要解决一个复杂的问题时，一个很有效的方法就是从一个简单的具体问题入手，寻找普遍的规律。

规律：首先选取数组中右上角的数字，如果该数字等一要查找的数字，查找过程结束；如果该数字大于要查找的数字，剔除这个数字所在的列；如果这个数字小于要查找的数字，剔除这个数字所在的行。

#include<iostream>

using std::cout;
using std::endl;

bool find_num(int *a,int num,int rows,int cols)
{
    bool result = false;

    if (a!=NULL&&rows>0&&cols>0)
    {
        int row = 0;
        int col = cols-1;

        while (row < rows&&col >=0)
        {
            if (a[row*cols + col ] == num)
            {
                result = true;
                break;
            }
            else if (a[row*cols + col]>num)
            {
                --col;
            }
            else
            {
                ++row;
            }
        }
    }

    return result;
    
    
}

void Test(char* testName,int *m,int rows,int cols,int num,bool expected)
{
    if (testName!=NULL)
    {
        cout << testName << endl;
    }

    bool result = find_num(m, num, rows, cols);

    if (result == expected)
    {
        cout << "passed" << endl;
    }
    else
        cout << "failed" << endl;
}

void Test1()
{
    int matrix[][4] = { { 1, 2, 8, 9 }, { 2, 4, 9, 12 }, { 4, 7, 10, 13 }, { 6, 8, 11, 15 } };

    Test("test1", (int*)matrix, 4, 4, 7, true);
}

void Test2()
{
    int matrix[][4] = { { 1, 2, 8, 9 }, { 2, 4, 9, 12 }, { 4, 7, 10, 13 }, { 6, 8, 11, 15 } };

    Test("test2", (int*)matrix, 4, 4, 20, false);
}

void Test3()
{
    

    Test("test3", NULL, 0, 0, 5, false);
}

int main()
{
    Test1();
    Test2();
    Test3();

    return 0;
}