Search for target value in matrix where every row & col is sorted
Perform 2 binary searches: 1 to find row, then another to find col
Time: O(log m + log n)
Space: O(1)
*/
class Solution {
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
int lowRow = 0;
int highRow = matrix.size() - 1;
while (lowRow < highRow) {
int mid = lowRow + (highRow - lowRow) / 2;
if (matrix[mid][0] == target) {
return true;
}
if (matrix[mid][0] < target && target < matrix[mid + 1][0]) {
lowRow = mid;
break;
}
if (matrix[mid][0] < target) {
lowRow = mid + 1;
} else {
highRow = mid - 1;
}
}
int lowCol = 0;
int highCol = matrix[0].size() - 1;
while (lowCol <= highCol) {
int mid = lowCol + (highCol - lowCol) / 2;
if (matrix[lowRow][mid] == target) {
return true;
}
if (matrix[lowRow][mid] < target) {
lowCol = mid + 1;
} else {
highCol = mid - 1;
}
}
return false;
}
};