[NewCoder 3] 二维数组中的查找

题目描述

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

 

来分析下，假设在数组中随便找一个数字 inner_number 来与 target 进行比较，如果 target > inner_number，那么我们需要在数组中剔除掉 inner_number 所在行的左边的数、 inner_number 所在列的上边的数以及左上部分的数；同理，如果 target < inner_number，那么我们需要在数组中剔除掉 inner_number 所在行的右边的数、 inner_number 所在列的下边的数以及右下部分的数。

小结一下，

如果 target 大了，那么剔除左上部分的数字

如果 target 小了，那么剔除右下部分的数字

这样一来，要表示数组中剩下的数字就比较蛋疼。如果可以一次剔除掉一行或者一列那就比较爽了，那么有没有这样的办法呢？当然，在一开始将 target 与右上角的数来比较，就可以做到一次剔除一行或者一列。同理，一开始将 target 与左下角的数比较也可以。

Why? 对于右上角的数字，如果 target 大了，那么就剔除左上部分的数字，上面没有东西了，所以就变成剔除该数字所在的行了；如果 target 小了，那么就剔除右下部分的数字，右边没有东西了，所以就变成剔除该数字所在的列了。左下角同理。

但是，左上角的数字与右下角的数字不可以。按左上角的数字来举例。如果 target 大了，那么剔除左上部分，特么左边与上边都没了，就只能剔除自己，这也太蛋疼了…当然 target 小了是好办，一下子把数字都剔除光了，返回 false 。

来看代码：

class Solution {
public:
    bool Find(vector<vector<int> > array,int target) {

        int maxRow = array.size() - 1;
        int minCol = 0;

        int curRow = 0;
        int curCol = array[0].size() - 1;

        // 从右上角开始查询，一次删除一行或者一列
        while (curRow <= maxRow && curCol >= minCol) {
            if (array[curRow][curCol] == target) {
                return true;
            } else if (array[curRow][curCol] < target) {
                curRow++;
            } else {
                curCol--;
            }
        }

        return false;
    }
};

顺带写个递归版本玩玩：

class Solution {
public:
	bool Find(vector<vector<int> > array,int target) {
        int maxRow = array.size() - 1;
        int minCol = 0;

        int curRow = 0;
        int curCol = array[0].size() - 1;

        return find_helper(array, target, curRow, curCol, maxRow, minCol);
    }

private:
    bool find_helper(vector<vector<int> > &array, const int target, int &curRow, int &curCol, const int maxRow, const int minCol) {
        if (curRow > maxRow || curCol < minCol) {
            return false;
        }

        if (array[curRow][curCol] == target){
            return true;
        } else if (array[curRow][curCol] < target) {
            return find_helper(array, target, ++curRow, curCol, maxRow, minCol);
        } else {
            return find_helper(array, target, curRow, --curCol, maxRow, minCol);
        }
    }

};