3 - 在有序二维数组中查找元素

题目描述：http://ac.jobdu.com/problem.php?pid=1384

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

---

解析：
由于每一行是递增的，每一列是递增的，

• 如果我们从左上角开始查找，那么当查找元素比左上元素大时，无法判断是应该向下、向右还是向右下；从右下也是同理。
• 但是如果我们从右上角开始查找，当查找元素比右上元素大时，那么第一行淘汰，只能向下；当查找元素比右上元素小时，那么最后一列淘汰，只能向左；从左下也是同理。

当遇到行列均有序时，除了考虑从左上角开始遍历时，应该有从右上或者左下开始遍历的新思路。

#include <iostream>
using namespace std;
// 根据简单案例，找规律
bool findIn2DArray(int** matrix, int rows, int cols, int number)
{
    if (matrix == NULL || rows <= 0 || cols <= 0)
        return false;
    int i = 0, j = cols-1; // 从右上角开始遍历，num小于右上值，左走；num大于右上值，下走（也可以将左下当比较元素）
    while (i < rows && j >= 0) {
        if (matrix[i][j] == number)
            return true;
        else if (matrix[i][j] > number)
            j--;
        else
            i++;
    }
    return false;
}
int main()
{
    int **matrix = new int* [3];
    for (int i = 0; i < 3; i++)
        matrix[i] = new int [3];
    int k = 0;
    for (int i = 0; i < 3; i++)
        for (int j = 0; j < 3; j++)
            matrix[i][j] = k++;
    cout << findIn2DArray(matrix, 3, 3, 10) << endl;
    cout << findIn2DArray(matrix, 3, 3, 0) << endl;
    cout << findIn2DArray(matrix, 3, 3, 8) << endl;
    cout << findIn2DArray(matrix, 3, 3, 3) << endl;
    cout << findIn2DArray(NULL, 3, 3, 3) << endl;
    cout << findIn2DArray(matrix, 0, 3, 3) << endl;
}

输出结果：
0
1
1
1
0
0
[Finished in 0.4s]