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

133 阅读2分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动

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

为了帮助大家更好地理解交互式问题,我们准备了一个样例 “答案”:leetcode-cn.com/playground/… 不是一个正确答案。

  • 示例 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个步骤

  1. 二分法找出峰值所在的下标,将数组划分为递增区间和递减区间
  2. 因为我们要找等于 target 最小 的下标 index 值,所以我们需要先在递增区间内找
  3. 然后再从递减区间内查找

代码

/**
 * // This is MountainArray's API interface.
 * // You should not implement it, or speculate about its implementation
 * interface MountainArray {
 *     public int get(int index) {}
 *     public int length() {}
 * }
 */
 
class Solution {
    public int findInMountainArray(int target, MountainArray mountainArr) {
        
        int l=0,r=mountainArr.length()-1;
        while(l<r){
            int mid=(r-l)/2+l;
            if(mountainArr.get(mid)<mountainArr.get(mid+1))
            {
                l=mid+1;
            }else r=mid;
        }
        int m=l;
        l=0;r=m;
        while(l<=r)
        {
             int mid=(r-l)/2+l;
            if(mountainArr.get(mid)==target)
            {
               return mid;
            }else if(mountainArr.get(mid)>target)
            {
                r=mid-1;
            }else l=mid+1;
        }
        l=m+1;r=mountainArr.length()-1;
        while(l<=r)
        {
             int mid=(r-l)/2+l;
            if(mountainArr.get(mid)==target)
            {
               return mid;
            }else if(mountainArr.get(mid)<target)
            {
                r=mid-1;
            }else l=mid+1;
        }
        return -1;
    }
}