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

Author: kamyu104

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

@@ -8,17 +8,18 @@ def findPeakGrid(self, mat):
         :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)))
+            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
+        left, right = 0, (max(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]
+        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]
+