1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
| #include <vector> using std::vector;
class Solution { private: bool isPeak(vector<int> &nums, int mid) { int n = nums.size(); if (mid != 0 && nums[mid] <= nums[mid - 1]) { return false; } if (mid != n - 1 && nums[mid] <= nums[mid + 1]) { return false; } return true; }
public: int findPeakElement(vector<int> &nums) { int n = nums.size(); int left = 0, right = n - 1, mid;
while (left <= right) { mid = left + (right - left) / 2; if (isPeak(nums, mid)) { return mid; }
if (nums[mid] < nums[mid + 1]) {
left = mid + 1; } else {
right = mid - 1; } }
return mid; } };
|