[LeetCode]1095. 山脉数组中查找目标值

题目

（这是一个 交互式问题 ）

给你一个 山脉数组 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

解题思路

1）首先找到峰值坐标 top，对于一个范围 [i, j]，我们可以先找到范围 [i, j] 中间连续的两个点 mid 与 mid + 1。如果 mountainArr.get(mid + 1) > mountainArr.get(mid)，那么可以知道峰值在范围 [mid + 1, j] 内；如果 mountainArr.get(mid + 1) < mountainArr.get(mid)，那么可以知道峰值在范围 [i, mid] 内。
2）根据峰值坐标将数组分为两部分，在[0,top]区间内数组单调递增，在[top+1, n-1]区间内数组单调递减。题目要求的是目标值的索引最小，所以先在左边寻找，如果没有找到，再在右边寻找，如果都没找到，则返回-1。
复杂度分析：
时间复杂度：O(logn)，我们进行了三次二分搜索，每次的时间复杂度都为 O(logn)。
空间复杂度：O(1)，只需要常数的空间存放若干变量。

代码

# """
# 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:
        n = mountain_arr.length()
        left = 0
        right = n-1
        top = self.findMountainTop(mountain_arr, 0, n-1)
        id1 = self.findInSortedArray(target, mountain_arr, 0, top)
        if id1 != -1:
            return id1
        else:
            return self.findInReversedArray(target, mountain_arr, top+1, n-1)

    def findMountainTop(self, mountain_arr: 'MountainArray', left: int, right: int):
        while left<right:
            mid = left + (right-left)//2
            if mountain_arr.get(mid)<mountain_arr.get(mid+1):
                left = mid+1
            else:
                right = mid
        return left

    def findInSortedArray(self, target: int, mountain_arr: 'MountainArray', left: int, right: int):
        while left <= right:
            mid = left + (right-left)//2
            if mountain_arr.get(mid) == target:
                return mid
            elif mountain_arr.get(mid) > target:
                right = mid-1
            else:
                left = mid+1
        return -1


    def findInReversedArray(self, target: int, mountain_arr: 'MountainArray', left: int, right: int):
        while left <= right:
            mid = left + (right-left)//2
            if mountain_arr.get(mid) == target:
                return mid
            elif mountain_arr.get(mid) > target:
                left = mid+1
            else:
                right = mid-1
        return -1