算法提高之LeetCode刷题数据结构和算法分析

山脉数组中查找目标值

2020-04-29  本文已影响0人  _阿南_

题目:

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

题目的理解:

在左边升序的数组中找target,如果没有找到则从右边的target进行查找。使用二分法查找。

python实现

class Solution:
    def findInMountainArray(self, target: int, mountain_arr: 'MountainArray') -> int:

        def binarySearch(mountain, target, l, r, key=lambda x: x) -> int:
            target = key(target)

            while l <= r:
                middle = (l + r) // 2
                cur = key(mountain.get(middle))

                if cur == target:
                    return middle
                elif cur < target:
                    l = middle + 1
                else:
                    r = middle - 1

            return -1


        l, r = 0, mountain_arr.length() - 1

        while l < r:
            middle = (l + r) // 2

            if mountain_arr.get(middle) < mountain_arr.get(middle + 1):
                l = middle + 1
            else:
                r = middle - 1

        mountainTop = l

        index = binarySearch(mountain_arr, target, 0, mountainTop)
        if index != -1:
            return index

        return binarySearch(mountain_arr, target, mountainTop, mountain_arr.length() - 1, key=lambda x: -x)

模仿最优解编写的,野蛮方法太多的if-else了,也就是没有正真的熟悉二分法的精髓。

想看最优解法移步此处

提交

哈哈

// END 活到老学到学,多读书读好书

上一篇下一篇

猜你喜欢

热点阅读