python: add problem 74 and unittest

Author: supercoderhawk

Python/sln_1_100/solution_71_80.py

@@ -7,6 +7,7 @@
 class Solution_71_80(object):
     def simplifyPath(self, path):
         """
+        71
         :type path: str
         :rtype: str
         """
@@ -37,6 +38,49 @@ def simplifyPath(self, path):
 
         return '/' + '/'.join(final_path_segments)
 
+    def searchMatrix(self, matrix: List[List[int]], target: int) -> bool:
+        """
+        74
+        :param matrix:
+        :param target:
+        :return:
+        """
+        if not matrix:
+            return False
+        if not matrix[0]:
+            return False
+        row = len(matrix) - 1
+        col = len(matrix[0]) - 1
+
+        def binary_search(func, l):
+            left = 0
+            right = l
+            if func(0) >= target:
+                return 0
+            elif func(l) <= target:
+                return l
+            else:
+                mid = (left + right) // 2
+                while left < mid:
+                    mid_val = func(mid)
+                    if mid_val < target:
+                        left = mid
+                    elif mid_val > target:
+                        right = mid
+                    else:
+                        return mid
+                    mid = (left + right) // 2
+
+            return left
+
+        row_index = binary_search(lambda i: matrix[i][0], row)
+        if matrix[row_index][0] == target:
+            return True
+        col_index = binary_search(lambda i: matrix[row_index][i], col)
+        if matrix[row_index][col_index] == target:
+            return True
+        return False
+
     def combine(self, n: int, k: int) -> List[List[int]]:
         """
         77

Python/sln_1_100/test_solution_71_80.py

@@ -15,6 +15,24 @@ def test_simplifyPath(self):
         self.assertEqual(self.sln.simplifyPath('///'), '/')
         self.assertEqual(self.sln.simplifyPath('/...'), '/...')
 
+    def test_search_martrix(self):
+        matrix = [
+            [1, 3, 5, 7],
+            [10, 11, 16, 20],
+            [23, 30, 34, 50]
+        ]
+        self.assertEqual(self.sln.searchMatrix(matrix, 3), True)
+        self.assertEqual(self.sln.searchMatrix(matrix, 7), True)
+        self.assertEqual(self.sln.searchMatrix(matrix, 11), True)
+        self.assertEqual(self.sln.searchMatrix(matrix, 50), True)
+        self.assertEqual(self.sln.searchMatrix(matrix, 8), False)
+        self.assertEqual(self.sln.searchMatrix(matrix, 100), False)
+        self.assertEqual(self.sln.searchMatrix(matrix, 23), True)
+        self.assertEqual(self.sln.searchMatrix(matrix, 0), False)
+        self.assertEqual(self.sln.searchMatrix(matrix, 1), True)
+        self.assertEqual(self.sln.searchMatrix([], 1), False)
+        self.assertEqual(self.sln.searchMatrix([[]], 1), False)
+
     def test_combine(self):
         ret = self.sln.combine(4, 2)
         self.assertEqual(ret, [[1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4]])