数组与矩阵---在数组中找到一个局部最小的位置

来源:互联网 发布:java线程池 关闭 编辑:程序博客网 时间:2024/06/05 21:15

【题目】

  定义局部最小的概念。arr长度为1时,arr[0]是局部最小。arr的长度为N(N>1)时,如果arr[0] < arr[1],那么arr[0]是局部最小;如果arr[N-1] < arr[N-2],那么arr[N-1]是局部最小;如果 0 < i < N-1,既有 arr[i] < arr[i-1],又有arr[i] < arr[i+1],那么arr[i]是局部最小。
  给定无序数组arr,已知arr中任意两个相邻的数都不相等。写一个函数,只返回arr中任意一个局部最小出现的位置即可。

【基本思路】

  本题可以使用二分查找做到时间复杂度O(logN),空间复杂度O(1),具体步骤如下:

  1. 如果arr长度为1或者,arr[0] < arr[1],返回0;如果arr[N-1] < arr[N-2],返回N-1

  2. 如果arr长度大于2且arr的左右两头都不是局部最小,令left = 1,right = N-2,然后开始二分查找:

    (1)令 mid = (left + right) // 2

    (2)如果arr[mid] > arr[mid-1],可知在arr[left…mid-1]上肯定存在局部最小值,令right = mid - 1

    (3)如果arr[mid] > arr[mid+1],可知在arr[mid+1…right]上肯定存在局部最小值,令left = mid + 1

    (4)如果不满足(2)和(3),则mid本身就是局部最小,返回mid即可

    (5)重复二分查找步骤,直到left > right

【代码实现】

def indexOfLocalMin(arr):    if arr == None or len(arr) == 0:        return -1    if len(arr) == 1 or arr[0] < arr[1]:        return arr[0]    if arr[-1] < arr[-2]:        return arr[-1]    left = 1    right = len(arr) - 2    while left <= right:        mid = (left + right) // 2        if arr[mid] > arr[mid-1]:            right = mid - 1        elif arr[mid] > arr[mid+1]:            left = mid + 1        else:            return mid
阅读全文
2 0