题目大意就是给定一个Vector arr,该动态数组一定在区间[0,x]和[x,arr.length]都单调递增,求x对应数组元素的下标。
方法一:利用山脉数组的特点,直接for循环过一遍找出arr[i]>arr[i+1]的i值即可。
时间复杂度:O(n),其中 n是数组arr的长度。我们最多需要对数组arr 进行一次遍历。
空间复杂度:O(1)。
class Solution {
public:
int peakIndexInMountainArray(vector<int>& arr) {
int n = arr.size();
int ans = -1;
for (int i = 1; i < n - 1;i++) {
if (arr[i] > arr[i + 1]) {
ans = i;
break;
}
}
return ans;
}
};
方法二:二分查找
记满足答案的i为k,则i<k时,arr[i]<arr[i+1];i>=k时,arr[i]>arr[i+1],那么我们就能用二分查找的方法找到k。
时间复杂度:O(logn),n为arr的长度,我们需要二分查找的次数是logn。
空间复杂度:O(1)。
class Solution {
public:
int peakIndexInMountainArray(vector<int>& arr) {
int n=arr.size();
int l=1,r=n-2,ans=0;
while(l<=r){
int mid=l+(r-l)/2;
if(arr[mid]>arr[mid+1]){
ans=mid;
r=mid-1;
}
else{
l=mid+1;
}
}
return ans;
}
};