给你一个满足下述两条属性的 m x n
整数矩阵:
- 每行中的整数从左到右按非严格递增顺序排列。
- 每行的第一个整数大于前一行的最后一个整数。
给你一个整数 target
,如果 target
在矩阵中,返回 true
;否则,返回 false
。
示例 1:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3 输出:true
示例 2:
输入:matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13 输出:false
提示:
m == matrix.length
n == matrix[i].length
1 <= m, n <= 100
-104 <= matrix[i][j], target <= 104
思路:
两次二分法,在二分法之前,先判断target与边界的关系,超出边界直接为false。然后从上到下使用二分法,把矩阵最后一列看成一维数组,查找target,定位到某一行,然后再对这一行的数组使用二分法查找,注意二分法区间不变性,本题我采用闭区间,也就是while里面是小于等于号。
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int m=matrix.size();
int n=matrix[0].size();
if(target<matrix[0][0]||target>matrix[m-1][n-1])
{
return false;
}
int up=0,down=m-1;
while(up<=down)
{
int middle=up+(down-up)/2;
if(target>matrix[middle][n-1])
{
up=middle+1;
}
else if(target<matrix[middle][n-1])
{
down=middle-1;
}
else
{
return true;
}
}
int row=up;
int left=0,right=n-1;
while(left<=right)
{
int middle=left+(right-left)/2;
if(target>matrix[row][middle])
{
left=middle+1;
}
else if(target<matrix[row][middle])
{
right=middle-1;
}
else
{
return true;
}
}
return false;
}
};