题目描述:
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
分析:
因为二维数组中,从左到右和从上到下都是递增排序,可以从右上角往左下角查找(或者从左下角往右上角查找),只需O(n)的时间复杂度,不需要遍历整个数组,那样时间复杂度是O(n^2)。从右上角往左下角查找,如果该位置的数比目标小,就往左移一位;比目标大,就往下移一位;相等则找到了,查找停止。
代码:
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int row = array.size();
int col = array[].size();
int i = , j = col - ; // 从右上角往左下查找
while(i < row && j >= ) {
if(array[i][j] == target) return true; // 找到了
if(array[i][j] > target) { // 往左查找
j--;
continue;
}
if(array[i][j] < target) { // 往下查找
i++;
continue;
}
}
return false;
}
};