牛客网剑指offer第一题,二维数组的查找,原地址在这里https://www.nowcoder.com/ta/coding-interviews
想要联练习的小伙伴可以去练习练习。
题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
这道题对python来说,只需要target in array就可以了,但是逻辑能力和数据结构能力得不到提示。下面记录一下思路,截图取自剑指offer这本书,还是挺推荐购买正版的。
假设数组是[[1,2,8,9],[2,4,9,12],[4,7,10,13],[6,8,11,15]],需要发现7这个数字,先从右上角9开始,如果9大于Target,则不考虑9这一列;如果9小于target,则不考虑9这一行,如下截图所示
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int row = array.size();
int col = array[0].size();
int col_i = col - 1;
int row_i = 0;
while(row_i < row && col_i >= 0){
if(array[row_i][col_i] > target)
col_i--;
else if(array[row_i][col_i] < target)
row_i++;
else
return true;
}
return false;
}
};