JAVA 牛牛

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

  • Apr 29, 2020
  • 0 views
  • 难度：困难
  • 本题涉及算法： 二分查找
  • 思路： 二分查找 找到峰顶 分割数组
  • 类似题型:

题目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。

提示：

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

方法一

解题思路

  • 搜索指定值 二分查找
  • 对于这种 可以分成左右数组来获取值
  • 先寻找峰值
class Solution(object):
    # 搜索指定值 二分查找
    # 对于这种 可以分成左右数组来获取值
    # 先寻找峰值
    def findInMountainArray(self, target, mountain_arr):
        res = [-1,-1]
        lo = 0
        hi = mountain_arr.length()
        top=0
        # 判断最高峰
        while lo<hi:
            mid = lo+ (hi-lo)//2
            if mountain_arr.get(mid) > mountain_arr.get(mid-1):
                lo = mid + 1
            else:
                hi = mid
        top = lo -1


        lo = 0
        hi = top

        # 左半
        while lo<hi:
            mid = lo+ (hi-lo)//2
            if mountain_arr.get(mid) == target:
                res[0] = mid
            if mountain_arr.get(mid) > target:
                hi = mid
            else:
                lo = mid + 1

        lo = top
        hi = mountain_arr.length()
        # 右半
        while lo<hi:
            mid = lo+ (hi-lo)//2
            if mountain_arr.get(mid) == target:
                res[1] = mid
            if mountain_arr.get(mid) > target:
                lo = mid + 1
            else:
                 hi = mid

        if res[0]==-1:
            return res[1]
        return res[0]

// java版会超时
public int findInMountainArray(int target, MountainArray mountainArr) {
        int lo =0,hi = mountainArr.length(),top=0;
        int[] res = new int[]{-1,-1};
        // 判断最高峰
        while(lo<hi) {
            int mid = lo+ (hi-lo)/2;
            if (mountainArr.get(mid]>mountainArr.get(mid-1))
                lo = mid+1;
            else if(mountainArr.get(mid]<mountainArr.get(mid-1))
                hi = mid;

        }
        top = lo-1;
        lo = 0;
        hi = top;

        // 左半
        while(lo<hi){
            int mid = lo+ (hi-lo)/2;
            if (mountainArr.get(mid)==target)
                res[0] = mid;
            if (mountainArr.get(mid)>target)
                hi = mid;
            else
                lo = mid+1;
        }

        hi = mountainArr.length();
        lo = top;
        // 右半
        while(lo<hi){
            int mid = lo+ (hi-lo)/2;
            if (mountainArr.get(mid)==target)
                res[1] = mid;
            if (mountainArr.get(mid)>target)
                lo = mid+1;
            else
                hi = mid;
        }
        if (res[0]==-1)
            return res[1];
        return res[0];

    }
Search