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。

提示:

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

解题思路

问题不是非常难,思路非常清晰。查找目标的问题不难想到二分法,但是使用二分法有一个前提条件就是数组必须有序。对于山脉数组来说,山峰的左边是递增的,山峰的右边是递减的。所以我们可以在山峰的左边和右边分别使用二分法查找目标。现在的问题就是如何判断山峰的位置?这个问题和Leetcode 941:有效的山脉数组(超详细的解法!!!)类似,所以我们可以通过该问题的解法去寻找山峰位置。但是本题有一点与之不同,在本题中,山峰数组一定存在。所以我们可以通过二分法查找,具体思路如下。

定义左右边界lrmid=(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

如有问题,希望大家指出!!!

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值