162. 寻找峰值
峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组
nums
,其中nums[i] ≠ nums[i+1]
,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。
你可以假设
nums[-1] = nums[n] = -∞
。示例 1:
输入: nums = [1,2,3,1] 输出: 2 解释: 3 是峰值元素,你的函数应该返回其索引 2。
示例 2:
输入: nums = [1,2,1,3,5,6,4] 输出: 1 或 5 解释: 你的函数可以返回索引 1,其峰值元素为 2; 或者返回索引 5, 其峰值元素为 6。
说明:
你的解法应该是 O(logN) 时间复杂度的。
解法一:
//时间复杂度O(n), 空间复杂度O(1)
class Solution {
public:
int findPeakElement(vector<int>& nums) {
for(int i = 0; i < nums.size() - 1; i++) {
if(nums[i] > nums[i + 1]) return i;
}
return nums.size() - 1;
}
};
//时间复杂度O(logn), 空间复杂度O(logn)
class Solution {
public:
int findPeakElement(const vector<int>& nums, int left, int right) {
if(left == right) return left;
int mid = left + (right - left) / 2;//[left, mid)长度 <= (mid, rihgt]长度
if(nums[mid] > nums[mid + 1]) return findPeakElement(nums, left, mid);
return findPeakElement(nums, mid + 1, right);
}
int findPeakElement(vector<int>& nums) {
return findPeakElement(nums, 0, nums.size() - 1);
}
};
解法一:
第一个峰值出现的位置就是序列首次出现递减趋势的位置(如果序列是非严格递增的,就说明还没有遇到峰值),如果整个序列都是非严格递增,返回最后一个元素的索引。
这个解法不满足时间要求。
解法二:
二分查找。在每个中点mid处,判断该点的局部增减性(题上说明了nums[i] != nums[i+1],也就是说一定只能是增或减),如果局部增,那向右半边继续查找,否则是局部减,向左半边查找。这个策略可以保证最后得到的一个元素一定是峰值。但是有一些边界细节要处理:
- 在每个递归层次内,需要对子序列[left, right]用mid进行分割,
mid = left + (right - left) / 2
这个操作会使mid左边的元素个数小于等于mid右边的元素个数。 - 终止条件是left == right,此时返回的left就是峰值元素的索引。
- 所以当
left < right
时,mid最小可以等于left,最大不会达到right,所以mid + 1一定不会越界,所以通过比较mid与mid+1这两个元素来判断局部增减性。如果nums[mid] > nums[mid + 1]
说明局部减,向左查找;否则向右查找。 - left和right每次更新的位置很重要,要保证当
left != right
时必定是left < right
。所以向左查找时就是right = mid
,向右查找是left = mid + 1
,这样保证了不会越界。见代码。
2019/12/31 12:27