LeetCode-Python-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 次调用的提交将被视为错误答案。此外，任何试图规避判题系统的解决方案都将会导致比赛资格被取消。

为了帮助大家更好地理解交互式问题，我们准备了一个样例 “答案”：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。

 

提示：

1. 3 <= mountain_arr.length() <= 10000
2. 0 <= target <= 10^9
3. 0 <= mountain_arr.get(index) <= 10^9

思路：

数组分为两部分有序，而且限定了调用次数，明示用二分法查找……

一共二分三次：

1. 找到peak的位置，peak即数组的最大值，以peak为分界线可以把原数组分成左右两个都有序的数组。

2. 在左侧有序数组里找target，因为如果target在两侧都出现，则需要返回下标小的那个。

3. 在右侧有序数组里找target。

class Solution(object):
    def findInMountainArray(self, target, mountain_arr):
        """
        :type target: integer
        :type mountain_arr: MountainArray
        :rtype: integer
        """
        # 得到数组长度
        n = mountain_arr.length()  
        #找山顶
        left, right = 0, n - 1
        while 1:
            mid = (left + right) // 2
            val = mountain_arr.get(mid)
            lval, rval = mountain_arr.get(mid - 1), mountain_arr.get(mid + 1)

            if val > lval and val > rval:
                peak = val
                peak_idx = mid
                break
            if val < rval:
                left = mid + 1
            else:
                right = mid
        if target == peak:
            return mid

        #找左侧
        left, right = 0, peak_idx - 1
        while left <= right:
            mid = (left + right) // 2
            val = mountain_arr.get(mid)
            if val == target:
                return mid
            elif val > target:
                right = mid - 1
            else:
                left = mid + 1
                
        #找右侧
        left, right = peak_idx + 1, n - 1
        while left <= right:
            mid = (left + right) // 2
            val = mountain_arr.get(mid)
            if val == target:
                return mid
            elif val > target:
                left = mid + 1
            else:     
                right = mid - 1
                
        return -1