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

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

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


解析:
由于每一行是递增的,每一列是递增的,

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

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

#include 
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]

你可能感兴趣的:(剑指offer)