给你一个 山脉数组 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次调用的提交将被视为错误答案。此外,任何试图规避判题系统的解决方案都将会导致比赛资格被取消。
思路:题目给的数组是前面部分为升序后面部分为降序的山脉数组,让我们在调用get()函数不超过100次的情况下找到指定的数字,可以采用二分搜索。
利用数组中最大的值来将数组分为左右两个部分,分别用二分查找即可。
/**
* // This is the MountainArray's API interface.
* // You should not implement it, or speculate about its implementation
* class MountainArray {
* public:
* int get(int index);
* int length();
* };
*/
class Solution {
public:
//二分查找,变量分别为:寻找的值,左右边界索引,排序方式( true:升序;false:降序 )
int binary( int target, int begin, int end, bool way, MountainArray &mountainArr ) {
int left = begin;
int right = end - 1;
int mid;
while( left <= right ) {
mid = left + (right - left) / 2;
int num = mountainArr.get(mid);
if( num == target ) return mid;
else if( num < target ) {
way == true ? left = mid + 1 : right = mid - 1;
}
else if( num > target ) {
way == true ? right = mid - 1 : left = mid + 1;
}
}
return -1;
}
int findInMountainArray(int target, MountainArray &mountainArr) {
int len = mountainArr.length(), max_index = -1;//山峰的索引
//寻找山峰
int left = 0, right = len - 1, mid;
while( left <= right ) {
int mid = left + (right - left) / 2;
if( mountainArr.get(mid) < mountainArr.get(mid + 1) ) left = mid + 1;
else right = mid - 1;
}
max_index = left;
//左右部分查找
int tmp = binary( target, 0, max_index, true, mountainArr);
return tmp == -1 ? binary( target, max_index, len, false, mountainArr) : tmp;
}
};