题目链接:74. 搜索二维矩阵 - 力扣(LeetCode)
在一个有序二维数组里面查找元素,同【LeetCode热题100】【矩阵】搜索二维矩阵 II-CSDN博客
如果用二分查找,时间复杂度是log(mn),但是可以实现时间复杂度为O(m+n)的,从右上角开始查找,如果当前元素比目标小,说明应该在下面范围,矩阵压扁,如果当前元素比目标大,说明在左边范围,矩阵压扁,这样最多只需要遍历m+n个元素可以确定元素是否存在
class Solution {
public:
bool searchMatrix(vector<vector<int> > &matrix, int target) {
int x = 0, y = matrix[0].size() - 1, n = matrix.size();
while (x < n && y >= 0) {
if (matrix[x][y] == target)
return true;
if (matrix[x][y] < target)
++x;
else
--y;
}
return false;
}
};