Update find-peak-element.py

kamyu104 · kamyu104 · commit 068432d6c2df · 2016-05-12T22:04:48.000+08:00
diff --git a/Python/find-peak-element.py b/Python/find-peak-element.py
@@ -28,7 +28,8 @@ def findPeakElement(self, nums):
         
         while left < right:
             mid = left + (right - left) / 2
-            if nums[mid - 1] <= nums[mid] >= nums[mid + 1]:
+            if (mid == 0 or nums[mid - 1] <= nums[mid]) and \
+               (mid == len(nums) - 1 or nums[mid + 1] <= nums[mid]):
                 return mid
             elif mid > 0 and nums[mid - 1] > nums[mid]:
                 right = mid
