题目:
峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums,找到峰值元素并返回其索引。数组可能蕴含多个峰值,在这种状况下,返回 任何一个峰值 所在位置即可。
你能够假如 nums[-1] = nums[n] = -∞ 。
你必须实现 工夫复杂度为 O(log n) 的算法来解决此问题。
链接: 力扣Leetcode—中级算法—排序和搜寻—寻找峰值.
示例 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(logN)的工夫复杂度要求,那么必定就是采纳二分法
二分法:
- 定义高下两个指针low,high,别离指向数组首位
- 定义两头指针mid=(low+high)/2咱们晓得二分法的完结标记是low>=high,当low==high的时候,算法必定完结
- 而后咱们每次判断两头值,如果两头值处于降落状态,那么能够必定两头值的右边有比两头值大的数,而数组中间又是十分小的数,所以峰值必定在右边,所以咱们二分法的下一次开始就是让high=mid
- 如果两头值处于回升状态,那么峰值必定就在两头值的左边,所以二分法的下一次开始就是让low=mid+1
- 你须要认真感触high=mid和low=mid+1这两个等式,琢磨轻微之处,能够发现low随着mid挪动,而high最多等于mid,认真琢磨他们的区别
全副Go代码如下:
package mainimport "fmt"//二分法func findPeakElement(nums []int) int { low, high := 0, len(nums)-1 for low < high { mid := (low + high) / 2 if nums[mid] > nums[mid+1] { high = mid } else { low = mid + 1 } } return low}func main() { a := []int{1, 2, 3, 1} fmt.Println(findPeakElement(a))}
提交截图:
这里我也写一下遍历:
全副Go代码如下:
package mainimport "fmt"//遍历func findPeakElement(nums []int) int { var max int n := len(nums) // 数组长度为1,那第一个就是最大,地位为0 if n == 1 { max = 0 } else if n == 2 { // 数组长度为2,比拟一下,如果第2个大于第1个,地位1为最大,其余都是地位0最大 if nums[1] > nums[0] { max = 1 } else { max = 0 } } else if n > 2 { // 数组长度大于2,遍历数组,如果有大于右边并且大于左边的,记下地位并且跳出遍历 for i := 1; i < n-1; i++ { if nums[i-1] < nums[i] && nums[i] > nums[i+1] { max = i break } else if nums[1] < nums[0] { // 如果没有合乎下面的要求,就剩下始终增的数组或者始终减的数组两种状况,只须要比拟第一和第二个数 max = 0 // 第一个数大,那地位0就是最大的 } else if nums[1] > nums[0] { max = n - 1 // 第二个数大,那地位n-1就是最大的 } } } return max}func main() { a := []int{1, 2, 3, 1} fmt.Println(findPeakElement(a))}
提交截图: