https://leetcode.com/problems/find-peak-element/
给定一个数组,数组具有相邻元素不相等的特性,即nums[i] != nums[i + 1]。现在定义峰值为大于左边、右边相邻元素的元素,返回任意一个峰值元素的下标。
nums[-1] = 负无穷, nums[nums.length] = 负无穷,即nums[0]只要大于右边元素nums[1],nums[nums.length - 1]只要大于左边元素nums[nums.length - 2],就是一个峰值元素。
一、问题分析
最简单的方式就是从下标0开始遍历一遍所有元素,当出现第一个峰值元素时,返回即可。这种方式返回的其实是第一个峰值元素的下标。但是,题目要求是说任意一个峰值元素即可。所以,这里还可以使用二分查找法。
我们可以把数组想象成几座小山相连,对于任意一个元素nums[mid],如果nums[mid] > nums[mid + 1],那么说明nums[mid]处在下坡路上。而既然是下坡路,说明nums[mid]的左边肯定至少有一个山顶(nums[mid]自己可能就是山顶)。既然如此,我们就可以剪枝掉nums[mid]右边部分,继续在nums[mid]左边部分查找,以达到缩小查找区间的目的。
二、代码实现
扫描一遍:
class Solution {
public int findPeakElement(int[] nums) {
if (nums.length == 1) {
return 0;
}
int i = 0;
for (i=0; i<nums.length; i++) {
if (i == 0) {
if (nums[i+1] < nums[i]) {
return i; //return 0;
}
continue;
}
if (i == nums.length - 1 && nums[i-1] < nums[i]) {
return i; //return nums.length - 1;
}
if (nums[i - 1] < nums[i] && nums[i + 1] < nums[i]) {
break;
}
}
return i;
}
}
二分查找法:
class Solution {
public int findPeakElement(int[] nums) {
int left = 0, right = nums.length - 1;
while (left < right) {
int mid = left + (right - left) / 2;
if (nums[mid] > nums[mid + 1]) {
//nums[mid]自己可能就是山顶,所以这里用right = mid
right = mid;
} else {
left = mid + 1;
}
}
//如果数组的长度为1,则直接跳过循环,跑到这里,直接返回0
return left;
}
}