1095：山脉数组中查找目标值（1827 分）

力扣第 142 场周赛第 3 题

题目

（这是一个 交互式问题 ）

你可以将一个数组 arr 称为 山脉数组 当且仅当：

• arr.length >= 3
• 存在一些 0 < i < arr.length - 1 的 i 使得：
  • arr[0] < arr[1] < ... < arr[i - 1] < arr[i]
  • arr[i] > arr[i + 1] > ... > arr[arr.length - 1]

给定一个山脉数组 mountainArr ，返回 最小 的 index 使得 mountainArr.get(index) == target。如果不存在这样的 index，返回 -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() <= 104
• 0 <= target <= 109
• 0 <= mountain_arr.get(index) <= 109

相似问题：

• 0852：山脉数组的峰顶索引（1181 分）
• 1671：得到山形数组的最少删除次数（1912 分）
• 2100：适合野炊的日子（1702 分）

分析

可以先二分查找到山顶坐标 mid（第一个满足 A[i]>A[i+1] 的 i)，然后 [0, mid], [mid, n-1] 区间都是单调的， 可以分别二分查找 target。

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

def findInMountainArray(self, target: int, mountain_arr: 'MountainArray') -> int:
    n = mountain_arr.length()
    self.__class__.__getitem__ = lambda self, i: mountain_arr.get(i)>mountain_arr.get(i+1)
    mid = bisect_left(self, True, 1, n-2)
    self.__class__.__getitem__ = lambda self, i: mountain_arr.get(i)>=target
    i = bisect_left(self, True, 0, mid)
    if mountain_arr.get(i) == target:
        return i
    self.__class__.__getitem__ = lambda self, i: mountain_arr.get(i)<=target
    j = bisect_left(self, True, mid, n-1)
    if mountain_arr.get(j) == target:
        return j
    return -1

32 ms