题目叙述
编写一个高效的算法来判断 m x n 矩阵中,是否存在一个目标值。该矩阵具有如下特性:
每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。
分析
本题所叙述的矩阵,整体都是有序的,即从左到右,从上到下都是有序的。所以对于这种从左到右,从上到下依次遍历的矩阵,可以当做一个一维的数组来进行遍历。该一维数组的起点是0, 终点是行数乘以列数 -1 。
然后进行求得中间的位置,对于该中间位置,如何确定其对应的行号和列号是一个问题,即中间索引如果是mid_index, 那么对应的行号为mid_index/列数,对应的列号为mid_index/列数。
代码
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
if(matrix.size()==0) return false;
//如果当前位置的元素大于target的话,那么往右和往下都要舍弃
int l = 0;
int r = matrix.size() * matrix[0].size() - 1;
while(l<=r){
int mid = l + (r-l)/2;
int row = mid / matrix[0].size();
int column = mid % matrix[0].size();
if(matrix[row][column] == target) return true;
else if(matrix[row][column]>target) r = mid - 1;
else l = mid + 1;
}
return false;
}
};