Modified Search a 2D Matrix.java

varunu28 · varunu28 · commit 6b4557d6efbb · 2020-10-16T05:56:30.000-07:00
diff --git a/Medium/Search a 2D Matrix.java b/Medium/Search a 2D Matrix.java
@@ -3,23 +3,23 @@ public boolean searchMatrix(int[][] matrix, int target) {
     if (matrix.length == 0 || matrix[0].length == 0) {
       return false;
     }
-    int i = 0;
-    int j = 0;
-    int rows = matrix.length;
-    int cols = matrix[0].length;
-    while (i < rows && j < cols) {
-      int num = matrix[i][j];
-      if (num == target) {
+    int rowIdx = matrix.length - 1;
+    int colIdx = matrix[0].length - 1;
+    while (rowIdx >= 0 && colIdx >= 0) {
+      int val = matrix[rowIdx][colIdx];
+      if (val == target) {
         return true;
       }
-      else {
-        if (i != rows - 1 && matrix[i + 1][j] <= target) {
-          i++;
-        }
-        else {
-          j++;
+      if (val >= matrix[rowIdx][0]) {
+        colIdx--;
+        if (colIdx < 0) {
+          colIdx = matrix[0].length - 1;
+          rowIdx--;
         }
       }
+      else {
+        rowIdx--;
+      }
     }
     return false;
   }
