原题
A peak element is an element that is greater than its neighbors.
Given an input array where num[i] ≠ num[i+1], find a peak element and return its index.
The array may contain multiple peaks, in that case return the index to any one of the peaks is fine.
You may imagine that num[-1] = num[n] = -∞.
For example, in array [1, 2, 3, 1], 3 is a peak element and your function should return the index number 2.
大意
找出数组中的峰值,即大于两边的数,假设num[-1] = num[n] = -∞,然后返回峰值的下标。要求使用对数复杂度来完成。
思路
如果使用遍历的方式,也能完成,但是时间复杂度是O(n),所以可以使用二分查找算法,来控制时间复杂度。
代码
//使用二分法查找算法来完成,这样复杂度就控制在对数复杂度
public class Solution {
public int findPeakElement(int[] nums) {
int i=0;
int j=nums.length-1;
if(nums.length==1)return 0;
while(i<j-1){
//求中间值
int n=(j-i)/2+i;
//如果中间值大于中间值后边的元素,则可以在前半部分找峰值
if(nums[n]>nums[n+1]){
j=n;
}
else{ i=n+1;
}
}
return nums[i]>nums[j]?i:j;
}
}
//使用循环遍历,但是时间复杂度不满足题目中的要求
public class Solution {
public int findPeakElement(int[] nums) {
if(nums.length==1)return 0;
int i = 1;
for (; i < nums.length-1; i++) {
if(nums[i-1]<nums[i]&&nums[i]>nums[i+1])
return i;
}
//考虑峰值在开始和结尾的元素
if(nums[nums.length-2]<nums[nums.length-1])
return nums.length-1;
if(nums[0]>nums[1])
return 0;
return i;
}
}