题目要求复杂度 logn,很容易就想到二分了。
然后,二分怎么移动 l,r 呢?其实也挺容易猜到的:如果 nums[mid] < nums[mid + 1],l = mid + 1,反之 r = mid - 1;
正确性说明:如果 nums[mid] < nums[mid + 1],那么右边一定是有峰值的,最坏情况也就是峰值在 nums.size() - 1 这个位置,由于题目说了 nums[n] 看做 -∞,所以必有答案;
class Solution { public: int findPeakElement(vector<int>& nums) { int l = 0, r = nums.size() - 1; while(l <= r) { int mid = (l + r) >> 1; bool tag1 = (mid == 0) || (nums[mid] > nums[mid - 1]); bool tag2 = (mid == nums.size() - 1) || (nums[mid] > nums[mid + 1]); if(tag1 && tag2) return mid; if(nums[mid] < nums[mid + 1]) { l = mid + 1; } else { r = mid - 1; } } return -1; } };