-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSearchIn2DMatrix.java
More file actions
70 lines (66 loc) · 2.34 KB
/
Copy pathSearchIn2DMatrix.java
File metadata and controls
70 lines (66 loc) · 2.34 KB
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
//PROBLEM
// 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.
public class SearchIn2DMatrix {
public static void main(String[] args) {
int[][] matrix = {
{1, 3, 5, 7},
{10, 11, 16, 20},
{23, 30, 34, 60}
};
int target=3;
Solution S1=new Solution();
boolean ans=S1.searchMatrix(matrix,target);
System.out.println(ans);
}
}
class Solution {
public boolean searchMatrix(int[][] matrix, int target) {
// for(int i=0; i<matrix.length; i++){
// for(int j=0; j<matrix[i].length; j++){
// if(matrix[i][j]==target){
// return true;
// }
// }
// }
// return false;
// the above solution works fine but its quite brute force and it did not pass all the test cases of lc and gave tle. as we can clearly see
// that we are using nested loop and checking each element. below is the otimized approach that i have implemented.
if(matrix==null || matrix.length==0 || matrix[0].length==0){
return false;
}
int s=0;
int e=matrix.length-1;
while(s<=e){
int mid=s+(e-s)/2;
if(matrix[mid][0]>target){
e=mid-1;
}else if(matrix[mid][0]<target){
s=mid+1;
}else{
return true;
}
}
if(e<0){
return false;
}
// now we know the row which might probably contains the target we will apply the binary search in that row. actually r is that row.
int start=0;
int end=matrix[r].length-1;
while(start<=end){
int mid=start+(end-start)/2;
if(matrix[r][mid]>target){
end=mid-1;
}else if(matrix[r][mid]<target){
s=mid+1;
}else{
return true;
}
}
return false;
}
}