add search 2d matrix

Author: LEUNGUU

src/search_2d_matrix.py

@@ -0,0 +1,29 @@
+from typing import List
+
+
+class Solution:
+    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
+        rows, cols = len(matrix), len(matrix[0])
+
+        top, bot = 0, rows - 1
+        while top <= bot:
+            row = (top + bot) // 2
+            if target > matrix[row][-1]:
+                top = row + 1
+            elif target < matrix[row][0]:
+                bot = row - 1
+            else:
+                break
+        if top > bot:
+            return False
+        row = (top + bot) // 2
+        left, right = 0, cols - 1
+        while left <= right:
+            middle = (left + right) // 2
+            if target > matrix[row][middle]:
+                left = middle + 1
+            elif target < matrix[row][middle]:
+                right = middle - 1
+            else:
+                return True
+        return False

tests/test_search_2d_matrix.py

@@ -0,0 +1,17 @@
+from search_2d_matrix import Solution
+
+testcase = [
+    {
+        "testcase": [[1, 3, 5, 7], [10, 11, 16, 20], [23, 30, 34, 60]],
+        "target": 3,
+        "result": True,
+    }
+]
+
+
+def test_search_2d_matrix():
+    for tc in testcase:
+        matrix = tc["testcase"]
+        target = tc["target"]
+        ans = Solution().searchMatrix(matrix, target)
+        assert ans == tc["result"]