-
Notifications
You must be signed in to change notification settings - Fork 215
/
Copy pathbinarySearch.cpp
36 lines (30 loc) · 1.21 KB
/
binarySearch.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
/*You are given an m x n integer matrix matrix with the following two properties:
Each row is sorted in non-decreasing order.
The first integer of each row is greater than the last integer of the previous row.
Given an integer target, return true if target is in matrix or false otherwise.
You must write a solution in O(log(m * n)) time complexity.
1.Input: matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3
Output: true
2.Input: matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13
Output: false*/
class Solution {
bool costumBinarySearch(const vector<vector<int>>& matrix, int target, int low, int high){
int size = matrix[0].size();
while(high - low > 1){
int mid = (high + low) / 2;
if(matrix[mid / size][mid % size] < target) {
low = mid + 1;
} else {
high = mid;
}
}
if(matrix[low / size][low % size] == target || matrix[high / size][high % size] == target){
return true;
}
return false;
}
public:
bool searchMatrix(vector<vector<int>>& matrix, int target) {
return costumBinarySearch(matrix, target, 0, matrix.size() * matrix[0].size() - 1);
}
};