-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSorted_matrix.java
More file actions
64 lines (64 loc) · 2.13 KB
/
Sorted_matrix.java
File metadata and controls
64 lines (64 loc) · 2.13 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
import java.util.Arrays;
public class Sorted_matrix {
public static void main(String[] args) {
int[][] matrix = {
{1, 2, 3, 4},
{5, 6, 7, 8},
{9, 10, 11, 12}
};
System.out.println(Arrays.toString(search(matrix,10)));
}
static int[] binarySearch(int[][] matrix, int row, int cStart, int cEnd, int target) {
while (cStart <= cEnd) {
int mid = cStart + (cEnd - cStart) / 2;
if (matrix[row][mid] == target) {
return new int[]{row, mid};
}
if (matrix[row][mid] < target) {
cStart = mid + 1;
} else {
cEnd = mid - 1;
}
}
return new int[]{-1, -1};
}
static int[] search(int[][] matrix, int target) {
int rows = matrix.length;
int cols = matrix[0].length;
if (rows == 1) {
return binarySearch(matrix, 0, 0, cols - 1, target);
}
int rStart = 0;
int rEnd = rows - 1;
int cMid = cols / 2;
while (rStart < (rEnd - 1)) {
int mid = rStart + (rEnd - rStart) / 2;
if (matrix[mid][cMid] == target) {
return new int[]{mid, cMid};
}
if (matrix[mid][cMid] < target) {
rStart = mid;
} else {
rEnd = mid;
}
}
if (matrix[rStart][cMid] == target) {
return new int[]{rStart, cMid};
}
if (matrix[rStart + 1][cMid] == target) {
return new int[]{rStart + 1, cMid};
}
if (target <= matrix[rStart][cMid - 1]) {
return binarySearch(matrix, rStart, 0, cMid - 1, target);
}
if (target <= matrix[rStart][cMid + 1] && target <= matrix[rStart][cols - 1]) {
return binarySearch(matrix, rStart, cMid + 1, cols - 1, target);
}
if (target <= matrix[rStart + 1][cMid - 1]) {
return binarySearch(matrix, rStart + 1, 0, cMid - 1, target);
}
else{
return binarySearch(matrix, rStart + 1, cMid + 1, cols - 1, target);
}
}
}