乐趣区

关于golang:Golang力扣Leetcode中级算法排序和搜索寻找峰值二分法

题目
峰值元素是指其值严格大于左右相邻值的元素。

给你一个整数数组 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 main

import "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 main

import "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))
}

提交截图

退出移动版