一开始以为一行的末尾小于下一行的开头,于是傻傻的写了两二分查找,所幸的是发现了自己在二分查找上确实不怎么熟练,巩固了一下知识。
这是一开始写的错误答案:
class Solution {
public:bool searchMatrix(vector<vector<int>>& matrix, int target) {int left=0;int right=matrix[0].size()-1;int record=0;while(left<=right){int middle=left+(right-left)/2;if(matrix[0][middle]>target) right=middle-1;else if(matrix[0][middle]<target) {record=left;left=middle+1;}else return 1;}left=0;right=matrix.size()-1;while(left+1<right){int middle=(left+right+1)/2;if(matrix[middle][record]>target) right=middle;else if(matrix[middle][record]<target) left=middle;else return 1;}if(matrix[left][record]==target||matrix[right][record]==target) return 1;return 0;}
};
记起来了二分查找中left和right不能只是等于mid,而是要在mid的基础上加或减,不然会死循环,并且跳出while的条件不能包含left==right,两者相等的情况下还要继续二分。
看答案学到了二分查找的逃课快速实现法:
class Solution {
public:bool searchMatrix(vector<vector<int>>& matrix, int target) {for(auto& row:matrix){auto it=lower_bound(row.begin(),row.end(),target);if(it!=row.end()&&*it==target) return 1;}return 0;}
};
lower_bound是一个二分查找的逃课函数,传入始末向量和target,就可以二分搜寻出最接近target的不小于target的元素,如果没有将返回end()。
答案还有一种很天才的思路:z字型查找,从右上开始搜索,如果比target小就删除最上行,比target大就删除最右行,非常高效。
class Solution {
public:bool searchMatrix(vector<vector<int>>& matrix, int target) {int n=matrix.size();int m=matrix[0].size();int x=0;int y=m-1;while(x<n&&y>=0){if(matrix[x][y]>target) y--;else if(matrix[x][y]<target) x++;else return 1;}return 0;}
};