力扣HOT100之二分查找:74. 搜索二维矩阵
这道题直接a了,我们可以参考上一道题:35.搜索插入位置的思路,详情见我的上一篇博客。将每一行的第一个元素当作一个数组中的元素,然后对这个数组进行二分查找,如果直接找到了target
,则直接返回true
,否则退出循环的时候left
指向应当插入的下标位置,举几个例子:
1.当输入为
[[1,3,5,7],[10,11,16,20],[23,30,34,60]]
target
为3
,则退出循环时left
指向元素10
所在的位置,此时我们应当去上一层(matrix[left - 1]
)寻找target
。
2.当输入为
[[1,3,5,7],[10,11,16,20],[23,30,34,60]]
target
为61,则退出循环时left
的值为3
,此时我们应当去上一层(matrix[left - 1]
,即matrix[2]
)寻找target
。
3.当输入为
[[1,3,5,7],[10,11,16,20],[23,30,34,60]]
target
为0,则退出循环时left
的值为0
,此时我们无法去上一层(matrix[-1]
,会发生溢出)寻找target
。所以我们需要判断一下退出二分查找后的left
是否大于0
,只有大于0
才进行进一步的搜索,当遍历完整层都没有找到时,返回false
,若找到则直接返回true
。
class Solution {
public:bool searchMatrix(vector<vector<int>>& matrix, int target) {int left = 0, right = matrix.size();int mid;while(left < right){mid = (left + right) / 2;if(matrix[mid][0] > target)right = mid;else if(matrix[mid][0] < target)left = mid + 1;else return true; }int i = 0;while(left > 0 && i < matrix[0].size()){if(matrix[left - 1][i] == target)return true;i++;}return false;}
};