❤️博客主页: 楚生辉
❤️系列专栏:【LeetCode刷题】
❤️一句短话: 坚持不懈,孜孜不倦
1.题目描述
符合下列属性的数组 arr
称为 山脉数组 :
arr.length >= 3
- 存在
i(0 < i < arr.length - 1)
使得:arr[0] < arr[1] < ... arr[i-1] < arr[i]
arr[i] > arr[i+1] > ... > arr[arr.length - 1]
给你由整数组成的山脉数组 arr ,返回任何满足arr[0] < arr[1] < ... arr[i - 1] < arr[i] > arr[i + 1] > ... > arr[arr.length - 1]
的下标 i 。
示例 1:
输入:arr = [0,1,0]
输出:1
示例 2:
输入:arr = [0,2,1,0]
输出:1
示例 3:
输入:arr = [0,10,5,2]
输出:1
示例 4:
输入:arr = [3,4,5,1]
输出:2
示例 5:
输入:arr = [24,69,100,99,79,78,67,36,26,19]
输出:2
提示:
3 <= arr.length <= 104
0 <= arr[i] <= 106
- 题目数据保证
arr
是一个山脉数组
进阶:很容易想到时间复杂度 O(n)
的解决方案,你可以设计一个 O(log(n))
的解决方案吗?
2.代码实现
- 方法一:使用最基本的循环遍历
class Solution {
public int peakIndexInMountainArray(int[] arr) {
// 找打处于山顶的那个下标
int i = 0;
while(arr[i] < arr[i+1]){
i ++ ;
}
return i;
}
}
- 方法二:使用二分查找法
可以看做寻找 arr[i + 1] 小于 arr[i]
区间内的左边界,也就是第一个arr[i + 1] 小于 arr[i]
的索引
class Solution {
public int peakIndexInMountainArray(int[] arr) {
// 找打处于山顶的那个下标
int left = 0;
int right = arr.length-1;
while(left <= right){
int mid = left + (right - left)/2;
if(arr[mid-1]<arr[mid] && arr[mid]>arr[mid+1]){
return mid;
}
if(arr[mid] > arr[mid+1]){
right = mid;
}else if(arr[mid] < arr[mid+1]){
left = mid;
}
}
return -1;
}
}
算法升级:
class Solution {
public int peakIndexInMountainArray(int[] arr) {
int n = arr.length;
int left = 1, right = n - 2, ans = 0;
while (left <= right) {
int mid = (left + right) / 2;
if (arr[mid] > arr[mid + 1]) {
ans = mid;
right = mid - 1;
} else {
left = mid + 1;
}
}
return ans;
}
}
或者
class Solution {
public int peakIndexInMountainArray(int[] arr) {
// 找打处于山顶的那个下标
int left = 1;
int right = arr.length-2;
int ans = 0;
while(left <= right){
int mid = left + (right - left)/2;
if(arr[mid] < arr[mid+1]){
left = mid + 1;
}else{
right = mid -1;
ans = mid;
}
}
return ans;
}
}