二维数组中的查找

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

如下面的二维数组就是每行、每列都递增排序。

1 2 8 9

2 4 9 12

4 7 10 13

6 8 11 15

按照剑指offer上面作者的解答，有如下规律：

首先选取数组右上角的数字。如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数字，提出这个数字所在的列；如果该数字小于要查找的数字，提出这个数字所在的行。也就是说，如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或一列。同理，也可以选择左下角。

代码如下：

bool find_(int *matrix, int rows, int cols, int number)
{
    bool flag = false;
    if(matrix != NULL && rows > 0 && cols > 0)
    {
        int row = 0;
        int col = cols - 1;
        while(row < rows && col >= 0)
        {
            int val = matrix[row * cols + col];
            if(val == number)
            {
                flag = true;
                break;
            }
            else if(val > number)
                col--;
            else
                row++;
        }
    }
    return flag;
}

第二种思路就是二分。

首先以最小的行上二分列，可以剔除掉所有比要查找数字大的列；然后在最大的列上二分行，可以剔除掉所有比要查找数字小的行。

代码如下：

bool find_bin(int *matrix, int rows, int cols, int number)
{
    if(matrix != NULL && rows > 0 && cols > 0)
    {
        int row = 0;
        int col = cols - 1;
        while(true)
        {
            int mid;

            int l = 0;
            int r = col;
            while(l <= r)
            {
                mid = (int)(l + r) / 2;
                if(matrix[row * cols + mid] == number)
                {
                    return true;
                }
                else if(matrix[row * cols + mid] < number)
                    l = mid + 1;
                else
                    r = mid -1;
            }
            col = l - 1;
            if(col < 0)
                break;

            l = row;
            r = rows - 1;
            while(l <= r)
            {
                mid = (int)(l + r) / 2;
                if(matrix[mid * cols + col] == number)
                {
                    return true;
                }
                else if(matrix[mid * cols + col] < number)
                    l = mid + 1;
                else
                    r = mid -1;
            }
            row = l;
            if(row >= rows)
                break;
        }
    }
    return false;
}

下面是测试代码：

int main()
{
    int matrix[] = {1, 2, 8, 9, 2, 4, 9, 12, 4, 7, 10, 13, 6, 8, 11, 15};
    cout<<find_(matrix, 4, 4, 7)<<endl;
    cout<<find_bin(matrix, 4, 4, 7)<<endl;
    return 0;
}