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

Author: kamyu104

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

@@ -21,4 +21,5 @@ def check(mat, x):
             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]
+        result = [left, next(i for i, x in enumerate(get_vec(mat, left)) if x == max_left)]
+        return result if len(mat) > len(mat[0]) else result[::-1]