leetcode1095. 山脉数组中查找目标值
(这是一个 交互式问题 )
给你一个 山脉数组 mountainArr,请你返回能够使得 mountainArr.get(index) 等于 target 最小 的下标 index 值。
如果不存在这样的下标 index,就请返回 -1。
何为山脉数组?如果数组 A 是一个山脉数组的话,那它满足如下条件:
首先,A.length >= 3
其次,在 0 < i < A.length - 1 条件下,存在 i 使得:
A[0] < A[1] < ... A[i-1] < A[i] A[i] > A[i+1] > ... > A[A.length - 1]
你将 不能直接访问该山脉数组,必须通过 MountainArray 接口来获取数据:
MountainArray.get(k) - 会返回数组中索引为k 的元素(下标从 0 开始) MountainArray.length() - 会返回该数组的长度
示例 1:
输入:array = [1,2,3,4,5,3,1], target = 3
输出:2
解释:3 在数组中出现了两次,下标分别为 2 和 5,我们返回最小的下标 2。
示例 2:
输入:array = [0,1,2,4,2,1], target = 3
输出:-1
解释:3 在数组中没有出现,返回 -1。
提示:
3 <= mountain_arr.length() <= 10000
0 <= target <= 10^9
0 <= mountain_arr.get(index) <= 10^9
方法:二分查找
思路:
这里面的山脉数组,可以认为是先升序,到峰值,再降序的过程,因此必须找到峰值,然后分割成两部分,在前边升序部分进行传统的二分查找,找到直接返回答案。若找不到,在后半部分降序进行修改之后的二分查找,找到返回答案,找不到返回-1。
如何找到峰值的坐标,是这里面的难点。
同样采用类似二分查找的方法,设置左右指针,求mid中间指针,以及mid+1。若mid+1对应的值比mid对应的值大,则说明最大值在[mid+1,right]中,否则说明最大值在[left,mid]下标范围内。直到左右指针相等,此时对应的下标即为峰值下标。
之后再[0,max]之间二分查找target,没有的话再在[max+1,right]二分查找(反向二分查找)
代码:
# """
# This is MountainArray's API interface.
# You should not implement it, or speculate about its implementation
# """
#class MountainArray:
# def get(self, index: int) -> int:
# def length(self) -> int:
class Solution:
def findInMountainArray(self, target: int, mountain_arr: 'MountainArray') -> int:
ans = -1
left = 0
right = mountain_arr.length() - 1
max_idx = self.findMaxIndex(mountain_arr,left,right) #找到最大值(峰顶)的下标
ans = self.binarysearch(mountain_arr,target,left,max_idx) #在左边升序部分二分找答案
if ans == -1: #如果没找到,在右边降序部分二分找答案
ans = self.binarysearchReverse(mountain_arr,target,max_idx+1,right)
return ans
def findMaxIndex(self,mountain_arr,left,right):#找到最大值的下标
if left == right:
return left
else:
mid = int((left+right)/2)
if mountain_arr.get(mid+1) - mountain_arr.get(mid) > 0: #mid+1的更大,说明最大值还在右边
return self.findMaxIndex(mountain_arr,mid+1,right)
else:
return self.findMaxIndex(mountain_arr,left,mid)
def binarysearch(self,mountain_arr,target,left,right): #正常的升序数组二分查找
if left == right:
if mountain_arr.get(left) == target:
return left
else:
return -1
mid = int((left+right)/2)
x = mountain_arr.get(mid)
if x == target:
return mid
elif x > target:
return self.binarysearch(mountain_arr,target,left,mid)
else:
return self.binarysearch(mountain_arr,target,mid+1,right)
def binarysearchReverse(self,mountain_arr,target,left,right): #降序的二分查找
if left == right:
if mountain_arr.get(left) == target:
return left
else:
return -1
mid = int((left+right)/2)
x = mountain_arr.get(mid)
if x == target:
return mid
elif x < target:
return self.binarysearchReverse(mountain_arr,target,left,mid)
else:
return self.binarysearchReverse(mountain_arr,target,mid+1,right)
结果:
![71a32b9f4024461700161e55a817f5ab.png](https://i-blog.csdnimg.cn/blog_migrate/d3c8c5147776e53aa0d180e95331f22f.png)