本文共 872 字,大约阅读时间需要 2 分钟。
题目:给定一个整型数组,从中找出某一个元素,使得该元素比其左右元素都大,求该元素在数组中的位置。4
代码一:直接求解
class Solution {public: int findPeakElement(vector & nums) { if(nums.size() == 1 || nums[0] > nums[1]) return 0; for(int x=1;xnums[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 & 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; }};
转载地址:http://nnrai.baihongyu.com/