https://leetcode.com/problems/find-peak-element/description/
题目:给定一个整型数组,从中找出某一个元素,使得该元素比其左右元素都大,求该元素在数组中的位置。4
代码一:直接求解
class Solution {
public:
int findPeakElement(vector<int>& nums) {
if(nums.size() == 1 || nums[0] > nums[1]) return 0;
for(int x=1;x<nums.size()-1;x++)
if(nums[x] > nums[x-1] && nums[x] > nums[x+1]) return x;
if(nums[nums.size()-1] > nums[nums.size()-2]) return nums.size()-1;
}
};
代码二:二分法
题目中已经说明,最左端和最右端元素均无限小,中间元素比两侧都要大,那么本题中一定存在一个峰元素。所以不管中间有多少波峰,只要找到峰元素,我们只需找到刚刚开始下降而未下降的位置。采用二分查找,查出这样一个位置即可,我们知道二分查找要比较的是 target 元素,本题的 target 元素是 mid 的后一个元素,即 nums[mid] 与 nums[mid+1] 进行比较:
class Solution {
public:
int findPeakElement(vector<int>& nums) {
int left = 0, right = nums.size() - 1;
while (left < right) {
int mid = (left + right) / 2;
if(nums[mid] < nums[mid + 1]) left = mid + 1;
else right = mid;
}
return left;
}
};