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

题目描述

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

image

 

来分析下,假设在数组中随便找一个数字 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);

        }

    }

    

};

你可能感兴趣的:(code)