Create find-a-peak-element-ii.py

Author: kamyu104

Python/find-a-peak-element-ii.py

@@ -0,0 +1,24 @@
+# Time:  O(min(n, m) * log(max(n, m)))
+# Space: O(1)
+
+class Solution(object):
+    def findPeakGrid(self, mat):
+        """
+        :type mat: List[List[int]]
+        :rtype: List[int]
+        """
+        def get_vec(mat, i):
+            return mat[i] if len(mat) < len(mat[0]) else (mat[j][i] for j in xrange(len(mat)))
+
+        def check(mat, x):
+            return max(get_vec(mat, x)) > max(get_vec(mat, x+1))
+
+        left, right = 0, (min(len(mat), len(mat[0]))-1)-1
+        while left <= right:
+            mid = left + (right-left)//2
+            if check(mat, mid):
+                right = mid-1
+            else:
+                left = mid+1
+        max_left = max(get_vec(mat, left))
+        return [left, next(i for i, x in enumerate(get_vec(mat, left)) if x == max_left)][::1 if len(mat) < len(mat[0]) else -1]