看到这道的第一时间就想到了二分,
因为:
- 数据的存储形式是数组,可以支持随机访问
- 数据有序排列,有内部规律
所以我们利用二分对这个过程进行优化
class Solution {
public:
bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
// 这道题肯定不可暴力遍历,看到有序查找,第一时间用二分。
// 大致思路是:单行的查找用二分加速
// 遍历是肯定要遍历的,但是要利用上数组的特点进行加速,比如我们要寻找10,那么第一行定位到下标为3,
//且不为target,那么我们可以确定的是,目标值一定不在第一行第三列右边的区域
if (matrix.empty() || matrix[0].empty()) return false;
int left = 0, right = matrix[0].size() - 1;
for (int i = 0; i < matrix.size(); i++) {
int mid = 0;
if (matrix[i][left] == target || matrix[i][right] == target) return true;
while(left < right) {
mid = (left + right + 1) >> 1;
if (matrix[i][mid] == target) return true;
if (matrix[i][mid] > target) right = mid - 1;
else left = mid;
}
if(matrix[i][mid] > target && (right = mid - 1) < 0) return false;
right = mid; left = 0;
}
return false;
}
};