坚持刷题第三天,备战实习。
一、题目
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。
请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
数据范围
二维数组中元素个数范围 [0,1000][0,1000]
样例
输入数组:
[
[1,2,8,9],
[2,4,9,12],
[4,7,10,13],
[6,8,11,15]
]
如果输入查找数值为7,则返回true,
如果输入查找数值为5,则返回false。
二、思路
我的想法就是用bfs的思想,因为数组是有序的,所以当前数一定比其右下角范围的数要小,所以如果当前数比目标target要小,则将当前数的右边和下面的序号加入q队列,如果相等则返回,大于则继续,不进行加队列的操作。
我的代码如下:
class Solution {
public:
bool searchArray(vector<vector<int>> array, int target) {
if(array.size()==0||array[0].size()==0) return false;
typedef pair<int,int> PII;
queue<PII> q;
q.push(make_pair(0,0));
bool st[1010][1010];
memset(st,true,sizeof st);
while(q.size()){
PII t = q.front();
q.pop();
if(array[t.first][t.second]==target){
return true;
}
else if(array[t.first][t.second]>target) {
st[t.first][t.second] = false;
continue;
}
else if(array[t.first][t.second]<target){
st[t.first][t.second] = false;
if(t.first + 1 < array.size()){
q.push(make_pair(t.first+1,t.second));
}
if(t.second + 1 < array[0].size()){
q.push(make_pair(t.first,t.second+1));
}
}
}
return false;
}
};
没想到题解从右上角开始判断,雀氏省掉了时间
class Solution {
public:
bool searchArray(vector<vector<int>> array, int target) {
if (array.empty() || array[0].empty()) return false;
int i = 0, j = array[0].size() - 1;
while (i < array.size() && j >= 0) {
if (array[i][j] == target) return true;
if (array[i][j] > target) j -- ;
else i ++ ;
}
return false;
}
};
作者:yxc
链接:https://www.acwing.com/solution/content/702/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
感觉智商又受到碾压,思维太固定了。