给你一个 山脉数组 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()
- 会返回该数组的长度
注意:
对 MountainArray.get
发起超过 100
次调用的提交将被视为错误答案。此外,任何试图规避判题系统的解决方案都将会导致比赛资格被取消。
示例 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
解题思路
问题不是非常难,思路非常清晰。查找目标的问题不难想到二分法,但是使用二分法有一个前提条件就是数组必须有序。对于山脉数组来说,山峰的左边是递增的,山峰的右边是递减的。所以我们可以在山峰的左边和右边分别使用二分法查找目标。现在的问题就是如何判断山峰的位置?这个问题和Leetcode 941:有效的山脉数组(超详细的解法!!!)类似,所以我们可以通过该问题的解法去寻找山峰位置。但是本题有一点与之不同,在本题中,山峰数组一定存在。所以我们可以通过二分法查找,具体思路如下。
定义左右边界l
和r
,mid=(l+r)//2
,判断arr[mid]>arr[mid+1]
是不是成立,如果成立,那么山峰在[0,mid]
这个区间中,我们在这个区间中继续查找;如果不成立,那么山峰在[mid+1,len(arr))
这个区间中,我们在这个区间中查找。
最后山峰左边查找目标和山峰右边查找目标使用的就是二分查找,不过山峰左边是递增,山峰右边是递减,所以需要分开考虑。如果不清楚二分法,请阅读Leetcode 二分法问题总结(超详细!!!)
class Solution:
def findInMountainArray(self, t: int, m: 'MountainArray') -> int:
p, r = 0, m.length() - 1
while p < r:
mid = (p + r) >> 1
if m.get(mid) > m.get(mid + 1):
r = mid
else:
p = mid + 1
i = self.bs(m, t, 0, p, True)
if i != -1:
return i
else:
return self.bs(m, t, p+1, m.length()-1, False)
def bs(self, m, t, l, r, asc):
while l < r:
mid = (l + r + 1) >> 1
if (asc and m.get(mid) <= t) or (not asc and m.get(mid) >= t):
l = mid
else:
r = mid - 1
if m.get(l) == t:
return l
return -1
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!