如题:
最简单的做法就是两层for循环查找。但是这样的时间复杂度为O(N2)显然不满足高效算法的要求,那么这个时候我们就需要使用二分法来解决问题:
首先思考一维数组二分法,无非就是
int low = 0;
int high = nums.size() - 1;
int mid = (high - low) / 2 + low;
while (low <= high) {
if (nums[mid] == target)return true;
else if (nums[mid] < target) low = mid + 1;
else if (nums[mid] > target) high = mid - 1;
}
return false;
那其实这种递增矩阵一样可以用同样的思路,如果把所有的行都放在第一行,那就是一个一维递增数组,如此使用二分法,只不过不同的是对于矩阵我们确定matrix中目标值需要使用一个小技巧就是**matrix[mid / n][mid % n]**仔细想一下 ,如果所有元素排成一行标上号,初始化high=m*n-1,low=0那么mid = (high - low) / 2 + low;mid在矩阵中的行是不是就是mid/n,一行最多n个元素,对应列数就是不超过n,就是mid%n,如此就可以用O(N)级别算法查找出特定值
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int m = matrix.size(), n = matrix[0].size();
int total = m * n - 1;
int first = 0;
while (first <= total) {
int mid = (total - first) / 2 + first;
if (matrix[mid / n][mid % n] == target)
return true;
else if (target > matrix[mid / n][mid % n])
first = mid + 1;
else if (target < matrix[mid / n][mid % n])
total = mid - 1;
}
return false;
}
};