题目:
峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 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。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-peak-element
说明:
你的解法应该是 O(logN) 时间复杂度的。
方法一
思路:
憨憨的方法,划分类别,分别处理,时间复杂度 O(N) 。
要注意题目中说的是返回一个峰值就可以啦。
代码如下:
/**
* 寻找峰值
* @param nums
* @return
*/
public int findpeekElement (int[] nums) {
int res = 0;
// 只有一个元素时
if (nums.length == 1) {
res = 0;
}
// 只有两个元素时
if (nums.length == 2) {
res = nums[0] > nums[1]? 0: 1;
}
// 三个元素及以上时,设置三个指针
for (int i = 0, j = 1, k = 2; k < nums.length; i++, j++, k++) {
// 边界情况
if (i == 0 && nums[i] > nums[j]) {
res = i;
continue;
}
// 中间情况
if (nums[j] > nums[i] && nums[j] > nums[k]) {
res = j;
continue;
}
// 边界情况
if (k == nums.length - 1 && nums[k] > nums[j]) {
res = k;
continue;
}
}
return res;
}
方法二
思路:
用二分法的思路。时间复杂度 O(logN).
代码如下:
/**
* 二分
* @param nums
* @return
*/
public int findpeekElement2 (int[] nums) {
int left = 0;
int right = nums.length -1;
int mid;
while (left < right) {
mid = left + (right - left) / 2;
// 中间和边界找到峰值时
if (mid == 0 && nums[mid] > nums[mid + 1] ||
mid == nums.length && nums[mid - 1] < nums[mid] ||
nums[mid] > nums[mid+1] && nums[mid] > nums[mid-1]) {
return mid;
}
// 移动left,right
if (nums[mid] > nums[mid + 1]) {
// 左边有峰值
right = mid;
} else {
left = mid + 1;
}
}
// 数组中只有一个元素时
return left;
}