题目来源:牛客网
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路
初次(逻辑错误):行列分别二分查找:从第0列中找到最接近的数,若非target,则从该指定行中二分查找target。
错误原因:并非每行之间的数分别都比比下一行中的每一个数小(any() < all()),所以答案不一定就在最第0列中接近的数的那一行。
正确思路:将左下角的数(或右上角同理)作为起始数。该数特征:本列最大,本行最小。
故:array[c][r] > target
时,只需column--
即可,同理if(array[c][r] < target) row--;
算法时间复杂度O(row+column) ,空间复杂度O(1)。
解题源码
#include <iostream>
#include <string>
#include <vector>
using namespace std;
class Solution
{
private:
public:
bool Find(int target, vector<vector<int>> array) {
int len_r = array[0].size();// lenth of row
int len_c = array.size(); // lenth of column
int res = array[len_c - 1][0]; //biggest of rows, smallest of columns
int r = len_r - 1;
int c = 0;
while (!(r < 0) && !(c > len_c - 1)) {
if (array[c][r] == target) return true;
else if (array[c][r] > target) r--;
else c++;
}
return false;
}
};
int main() {
vector<vector<int>> array;
vector<int> a;
vector<int> b;
a.push_back(2);
a.push_back(4);
a.push_back(6);
a.push_back(8);
b.push_back(3);
b.push_back(5);
b.push_back(7);
b.push_back(9);
array.push_back(a);
array.push_back(b);
Solution s;
int que = 5;
if (s.Find(que, array)) cout << "Find " << que << "!" << endl;
else cout << que << " doesn't exist." << endl;
return 0;
}