剑指Offer–二维数组的查找
题目:
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路:
题目给定二维数组按行递增,按列递增。
1. (错误)有一种很自然的想法就是分别按行和列定位,细想一下这是不行的:例如按行定位时,目标元素大于第5行行首元素,小于第6行行首元素,仅仅只能确定元素在第0行到第6行,并不能锁定在某行;并且有可能该元素大于所有的行首元素,无法确定具体信息,列也是同理。
2. (错误)从左上角开始遍历,若目标元素大于当前元素则可向右查找,也可向下查找,显然行不通。
3. (正确)从左下角开始遍历,当前元素是该列最大值,因此若目标元素大于当前元素,则向右继续查找;若目标元素小于当前元素,则向上查找。这样就保证了仅有一条查找路径。
4. (正确)同理,我们可以从右上角开始查找,道理同3。
注意:并没说二维数组是一个矩阵,也就是说并不是规则的正方形,每行都有相同元素值;很可能出现[[]],[[1,2,3],[],[]]类似情况,因此需要考虑数组越界的问题。
代码如下:
#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
using namespace std;
bool Find(int target,vector<vector<int>> array) {
if(array.size()==0) return false;
int row = array.size()-1;
int col = 0;
while(array[row].size()==0) row--;
if(row<0) return false;
int current = array[row][col];
while(current!=target) {
if(target<current) row--;
if(target>current) col++;
if(row<0||col>=array[row].size()) break;
current = array[row][col];
}
return (target==current)? true:false;
}
int main()
{
vector<vector<int>> v = {{}};
cout<<Find(1, v)<<endl;
return 0;
}