leetcode每日一题 山脉数组中查找目标值

leetcode每日一题 山脉数组中查找目标值

题目描述

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

解题思路

如果是一个有序数组，那么我们很容易就可以通过二分查找解决这个问题。那这个题目，如果我们能够找到这个最大值，就可以把数组分成升序和降序两部分。

对于一个区间$[i,j]$，令$mid=\frac{i+j}{2}$，如果$mid>mid-1$（这里的$mid$代表索引是$mid$的值），说明最大值在$[mid,j]$之间，反之则说明在$[i,mid]$之间。重复这个步骤，直到只剩两个元素，再比较这两个元素，就能得到最大值和它的索引。

接下来就可以通过二分查找查找目标值了，先在升序数组查找，如果没有就在降序数组查找。

贴出代码：

class Solution {
    public int findInMountainArray(int target, MountainArray mountainArr) {
        int lo = 0, hi = mountainArr.length() - 1;
        while (lo + 1 < hi) { // lo+1的原因是因为至少有3个元素
            int mid = lo + (hi - lo) / 2;
            int midVal = mountainArr.get(mid);

            if (midVal > mountainArr.get(mid - 1)) {
                lo = mid;
            } else {
                hi = mid;
            }
        }

        int peakIdx = mountainArr.get(lo) > mountainArr.get(hi) ? lo : hi;

        int res = binSearch(mountainArr, 0, peakIdx, target, true);
        return res == -1 ? binSearch(mountainArr, peakIdx + 1, mountainArr.length() - 1, target, false) : res;
    }

    private int binSearch(MountainArray mountainArr, int lo, int hi, int target, boolean bool) {
        while (lo <= hi) {
            int mid = lo + (hi - lo) / 2;
            int midVal = mountainArr.get(mid);
        

            if (midVal == target) {
                return mid;
            }

            if (midVal < target) {
                lo = bool? mid + 1 : lo;
                hi = bool? hi : mid - 1;
            } 
            if (midVal > target) {
                lo = bool? lo : mid + 1;
                hi = bool? mid - 1 :hi;
            }
        }
        return -1;
    }
}