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

Difficulty
Hard
Tags
二分查找
URL
https://leetcode-cn.com/problems/find-in-mountain-array/
Star
（这是一个 交互式问题 ）
给你一个 山脉数组 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 次调用的提交将被视为错误答案。此外，任何试图规避判题系统的解决方案都将会导致比赛资格被取消。
为了帮助大家更好地理解交互式问题，我们准备了一个样例 “答案”：https://leetcode-cn.com/playground/RKhe3ave，请注意这 不是一个正确答案。
示例 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
通过次数20,456提交次数54,936

法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:
        length = mountain_arr.length()
        if mountain_arr.get(0) > target and mountain_arr.get(length-1) > target:
            return -1

        left, right =  0, length-1
        # 先找到山脉顶峰
        while left < right:
            mid = left + (right-left) // 2
            temp = mountain_arr.get(mid)
            temp_2 = mountain_arr.get(mid+1)
            if temp > temp_2:
                right = mid
            elif temp < temp_2:
                left = mid + 1

        peakNum = mountain_arr.get(left)
        if peakNum < target:
            return -1
        print(left)
        # 先找左半段
        res = self.binarySearch(target, mountain_arr, 0, left, True)
        if res != -1:
            return res
        # 查找右半段
        res = self.binarySearch(target, mountain_arr, left, length-1, False)
        return res
        
        
    def binarySearch(self, target, mountain_arr, l, r, flag):
        # 通过flag来判断是查找左半段还是右半段
        while l <= r:
            mid = l + (r-l) // 2
            temp = mountain_arr.get(mid)
            # 左半段查找，为递增区间查找
            if flag:
                if temp < target:
                    l= mid + 1
                elif temp > target:
                    r = mid - 1
                else:
                    return mid
            # 右半段查找，为递减区间查找
            else:
                if temp < target:
                    r = mid - 1
                elif temp > target:
                    l= mid + 1
                else:
                    return mid
        return -1